modification de app.py pour la selection du model

app.py

@@ -14,14 +14,15 @@ app = FastAPI()
 
 class PromptRequest(BaseModel):
     prompt: str
+    model: str = "gemini/gemini-1.5-pro"  # Valeur par défaut
 
 @app.post("/")
 async def predict(request: PromptRequest):
     try:
         response = litellm.completion(
-            model=model_name, 
+            model=request.model, 
             messages=[{"role": "user", "content": request.prompt}]
         )
-        return {"response": response.choices[0].message.content}
+        return {"response": response.choices[0].message.content, "model": request.model}
     except Exception as e:
-        return {"error": str(e)}
+        return {"error": str(e)}