Update main.py

main.py

@@ -58,7 +58,7 @@ def read_root(request: Request, input_data: InputData):
 
     history = []  # Puoi definire la history se necessario
     generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
-    return generated_response
+    return {"response": generated_response}
 
 @app.post("/Immagine")
 def generate_image(request: Request, input_data: InputImage):
@@ -77,7 +77,7 @@ def generate_image(request: Request, input_data: InputImage):
     with open(image_url, 'rb') as img_file:
         img_binary = img_file.read()
         img_base64 = base64.b64encode(img_binary).decode('utf-8')
-    return img_base64
+    return {"response": img_base64}    
 
 @app.get("/")
 def read_general():