Update api.py

Added dynamic support for new models

api.py

@@ -8,6 +8,7 @@ class PromptRequest(BaseModel):
     text: str
     persona: str = "You are a knowledgable and helpful assistant."
     max_tokens: int = 500
+    model: str = "granite4:latest"
 
 # --- NEW: Add a Root Route (The Health Check) ---
 @app.get("/")
@@ -20,7 +21,7 @@ def generate(req: PromptRequest):
         res = requests.post(
             "http://localhost:11434/api/generate",
             json={
-                "model": "granite4:latest", 
+                "model": req_model, 
                 "prompt": f"{req.persona}\n\n{req.text}",
                 "stream": False,
                 "options": {