Update app.py

app.py

@@ -14,6 +14,11 @@ 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')
+status = f"Status{model.status}")
+
 class ChatRequest(BaseModel):
     message: str
     system_message: str = "You are a friendly Chatbot."
@@ -39,6 +44,7 @@ async def chat(request: ChatRequest):
             top_p=request.top_p,
         )
 
-        return {"response": response.choices[0].message.content}
+        # return {"response": response.choices[0].message.content}
+        return {"response": status}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))