App.py Update

Dockerfile

@@ -28,4 +28,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python", "app.py"]
+CMD ["python", "-u", "app.py"]

app.py

@@ -10,7 +10,6 @@ def interact(message: str, history: list):
         "role": "user",
         "content": message
     }
-
     chat_history = [msg for msg in history]
     chat_history.append(message_dct)
 
@@ -33,12 +32,11 @@ interface = gr.ChatInterface(
     description="Model: Microsoft Phi-2 (2.7B params)"
 )
 
-if __name__ == "__main__":
-    print("\n\nStarting Ollama...\n\n")
-    subprocess.Popen(["ollama", "serve"])
-    time.sleep(10)
-    print("\n\nOllama started successfully!!\n\n\n\nTesting...\n\n")
-    subprocess.run(["ollama", "pull", model_id])
-    time.sleep(5)
-    print("\n\nMicrosoft Phi-2 started successfully!!\n\n")
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+print("\n\nStarting Ollama...\n\n")
+subprocess.Popen(["ollama", "serve"])
+time.sleep(10)
+print("\n\nOllama started successfully!!\n\n\n\nTesting...\n\n")
+subprocess.run(["ollama", "pull", model_id])
+time.sleep(5)
+print("\n\nMicrosoft Phi-2 started successfully!!\n\n")
+interface.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,2 +1,2 @@
 ollama
-gradio
+gradio==5.14.0