Update main.py

main.py

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