Update main.py

main.py

@@ -7,7 +7,7 @@ from gradio_client import Client
 import base64
 import requests
 import os
-import subprocess
+import socket
 
 app = FastAPI()
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -54,6 +54,10 @@ 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/")
@@ -67,7 +71,7 @@ def generate_image():
         453666937,
         fn_index=0
     )
-    image_url = subprocess.run(["docker", "ps", "--format", "{{.Names}}"], capture_output=True, text=True) + os.path.abspath(os.path.dirname(__file__)) + result
+    image_url = get_ip_address() + result
     return {"response": image_url}
 
 @app.get("/Test")