Upload 2 files

Dockerfile

@@ -14,8 +14,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN apt-get update && apt-get install -y curl
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Pre-download the llama3 model
-RUN ollama pull llama3
+# Pre-download the llama3.2:1b model and debug
+RUN ollama pull llama3.2:1b && ollama list
 
 # Copy the app code
 COPY app.py .

app.py

@@ -51,8 +51,8 @@ def setup_client(provider):
         client = instructor.from_openai(
             OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama"), mode=instructor.Mode.JSON
         )
-        model = "llama3"
-        display_model = "Ollama (llama3)"
+        model = "llama3.2:1b"  # Updated to use the smaller model
+        display_model = "Ollama (llama3.2:1b)"
     else:
         st.error(f"Unsupported provider: {provider}")
         return None, None, None