App update

Dockerfile

@@ -12,8 +12,6 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-RUN ollama serve
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 from ollama import chat, ChatResponse
 import subprocess
+import time
+
+model_id = "deepseek-r1:1.5b"
 
 def interact(message: str, history: list):
     message_dct = {
@@ -12,7 +15,7 @@ def interact(message: str, history: list):
     chat_history.append(message_dct)
 
     response: ChatResponse = chat(
-        model="deepseek-r1:1.5b",
+        model=model_id,
         messages=chat_history,
         stream=True
     )
@@ -39,9 +42,16 @@ def interact(message: str, history: list):
 interface = gr.ChatInterface(
     fn=interact,
     type="messages",
-    title="Deepseek-R1 Chat Interface"
+    title="Deepseek-R1 Chat Interface",
+    description="Model: Deepseek R1: 1.5B params"
 )
 
 if __name__ == "__main__":
-    subprocess.run(["ollama", "run", "deepseek-r1:1.5b"])
-    interface.launch()
+    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\nDeepseek-R1 started successfully!!\n\n")
+    interface.launch(server_name="0.0.0.0", server_port=7860)