Update render.py

render.py

@@ -9,7 +9,7 @@ import io
 import base64
 import io
 import os
-
+import gradio as gr
 server_address = os.environ.get("URL_API")
 json_data=os.environ.get("JSON_API")
 client_id = str(uuid.uuid4())
@@ -31,6 +31,7 @@ def get_history(prompt_id):
         return json.loads(response.read())
 
 def get_images(ws, prompt):
+    progress=gr.Progress(track_tqdm=True)
     prompt_id = queue_prompt(prompt)['prompt_id']
     output_images = {}
     last_reported_percentage = 0
@@ -72,6 +73,7 @@ def pil_to_base64(image):
     return f"data:image/png;base64,{base64_string}" 
 
 def generate_images(positive_prompt, image):
+    
     ws = websocket.WebSocket()
     ws_url = f"ws://{server_address}/ws?clientId={client_id}"
     ws.connect(ws_url)