Update main.py

main.py

@@ -4,6 +4,9 @@ from huggingface_hub import InferenceClient
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
+class InputData(BaseModel):
+    input: str
+
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -12,11 +15,12 @@ def format_prompt(message, history):
   prompt += f"[INST] {message} [/INST]"
   return prompt
 
-@app.get("/Genera")
-def read_root(input):
+@app.post("/Genera")
+def read_root(request: Request, input_data: InputData):
+    input_text = input_data.input
     history = []  # Puoi definire la history se necessario
-    generated_response = generate(input, history)
-    return {"response": generated_response}  # Restituisci la risposta generata come JSON
+    generated_response = generate(input_text, history)
+    return {"response": generated_response}
 
 @app.get("/")
 def read_general():