Update main.py

main.py

@@ -3,6 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware  # Importa il middleware CORS
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 from datetime import datetime
+from gradio_client import Client
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -49,7 +50,7 @@ def read_root(request: Request, input_data: InputData):
     generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
     return {"response": generated_response}
 
-@app.get("/Immagine")
+@app.post("/Immagine")
 def generate_image():
     client = Client("https://openskyml-fast-sdxl-stable-diffusion-xl.hf.space/--replicas/545b5tw7n/")
     result = client.predict(
@@ -62,7 +63,7 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    return result
+    return {"response": result}
 
 @app.get("/")
 def read_general():