wip

Dockerfile

@@ -19,6 +19,3 @@ RUN apt-get update && \
     apt-get install -y openjdk-17-jre-headless
 
 CMD ["python", "gradio_app.py"]
-
-# sudo docker build -t document-translator .
-# docker run -p 7860:7860 --rm -it document-translator

gradio_app.py

@@ -2,10 +2,11 @@ import gradio as gr
 from src.translate_any_doc import translate_document
 from src.salamandraTA7b_translator import SalamandraTA7bTranslator
 from src.aligner import Aligner
+import os
 
 config_folder = 'fast_align_config'
 temp_folder = 'tmp'
-hf_token = ""
+hf_token = os.getenv('HF_TOKEN')
 
 translator = SalamandraTA7bTranslator(hf_token)
 

readme.md

@@ -29,4 +29,4 @@ Class that uses huggingface's demo.
 ## Docker
 
 sudo docker build -t document-translator .
-docker run -p 7860:7860 --rm -it document-translator
+docker run -p 7860:7860 -e HF_TOKEN=your_token_here --rm -it document-translator