Update app.py

app.py

@@ -15,8 +15,9 @@ if hf_token:
 else:
     raise ValueError("HF_TOKEN environment variable not set. Please add it as a secret in your Hugging Face Space.")
 
-# model = ollama.pull('llama3.2')
-model_name = 'hf.co/bartowski/Llama-3.2-3B-Instruct-GGUF:Q4_K_M'
+model = ollama.pull('llama3.2')
+# model_name = 'hf.co/bartowski/Llama-3.2-3B-Instruct-GGUF:Q4_K_M'
+
 model = ollama.pull(model_name)
 
 class ChatRequest(BaseModel):
@@ -37,7 +38,7 @@ async def chat(request: ChatRequest):
         start_time = time.time()  # Start measuring time
 
         messages = [
-            {'role': 'system', 'content': request.system_message}
+            {'role': 'system', 'content': request.system_message},
             {'role': 'user', 'content': request.message}
         ]