dockerfix

AstraBert · Apr 5, 2024 · 5c97eb6 · 5c97eb6
1 parent 5a11aab
commit 5c97eb6
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 11 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -29,16 +29,9 @@ ARG version="v0.1.0"
 # Echo version
 RUN echo "Started everything-rag ${version}"
 
-# Set default model and task
-ENV model="google/flan-t5-base"
-
-ENV task="text2text-generation"
-
-# Run script to install and build the model for the first time
-RUN python3 utils.py -m ${model} -t ${task}
-
 # Expose the port that the application will run on
 EXPOSE 7860
 
 # Set the entrypoint with a default command and allow the user to override it
-ENTRYPOINT ["python3", "chat.py", "-m", "${model}", "-t", "${task}"]
+ENTRYPOINT ["python3", "chat.py"]
+
diff --git a/docker/__pycache__/utils.cpython-310.pyc b/docker/__pycache__/utils.cpython-310.pyc
diff --git a/docker/chat.py b/docker/chat.py
@@ -73,6 +73,6 @@ def bot(history):
     gr.ClearButton(chatbot)
 demo.queue()
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", share=False)
 
 
diff --git a/docker/requirements.txt b/docker/requirements.txt
@@ -4,7 +4,6 @@ pypdf==3.17.4
 sentence_transformers==2.2.2
 chromadb==0.4.22
 gradio
-torch
 transformers
 trl 
 peft