Update app.py

app.py

@@ -4,7 +4,8 @@ import json
 
 # Define the URL for the local Ollama API and the model name
 OLLAMA_API_URL = "http://localhost:11434/api/generate"
-MODEL_NAME = "gemma-unsloth"  # This must match the name used in `ollama create` in run.sh
+# This must match the name used in `ollama pull` in Dockerfile
+MODEL_NAME = "hf.co/unsloth/gemma-3-4b-it-qat-GGUF:Q4_K_M"
 
 def generate_text(prompt, max_new_tokens=256, temperature=0.7):
     """