Update main.py

main.py

@@ -3,8 +3,6 @@ from fastapi.middleware.cors import CORSMiddleware  # Importa il middleware CORS
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 from datetime import datetime
-from fastapi.responses import JSONResponse
-import base64
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -53,7 +51,6 @@ def read_root(request: Request, input_data: InputData):
 
 @app.get("/Immagine")
 def generate_image():
-    # Chiamata al servizio gradio_client per ottenere l'immagine
     client = Client("https://openskyml-fast-sdxl-stable-diffusion-xl.hf.space/--replicas/545b5tw7n/")
     result = client.predict(
         "a giant monster hybrid of dragon and spider, in dark dense foggy forest ",
@@ -65,16 +62,7 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    
-    # Assume che 'result' sia l'immagine restituita come BASE64
-    image_base64 = result.get("data")  # Assumi che 'data' contenga l'immagine come BASE64
-    
-    if image_base64:
-        # Ritorna la risposta come JSON contenente il BASE64 dell'immagine
-        return JSONResponse(content={"image_base64": image_base64})
-    else:
-        # Nel caso in cui non si riesca a ottenere l'immagine
-        return JSONResponse(content={"message": "Impossibile ottenere l'immagine"}, status_code=500)
+    return result
 
 @app.get("/")
 def read_general():