Update main.py

main.py

@@ -54,10 +54,6 @@ 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}
 
-def get_ip_address():
-    ip_address = socket.gethostbyname(socket.gethostname())
-    return ip_address
-
 @app.get("/Immagine")
 def generate_image():
     client = Client("https://openskyml-fast-sdxl-stable-diffusion-xl.hf.space/--replicas/545b5tw7n/")
@@ -71,8 +67,11 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    image_url = get_ip_address() + result
-    return {"response": image_url}
+    image_url = result
+    with open(image_url, 'rb') as img_file:
+        img_binary = img_file.read()
+        img_base64 = base64.b64encode(img_binary).decode('utf-8')
+    return {"response": img_base64}
 
 @app.get("/Test")
 def generate_image():