Update app.py

app.py

@@ -19,14 +19,6 @@ repos = [
     "hsuwill000/Fluently-v4-LCM-openvino_3",
     "hsuwill000/Fluently-v4-LCM-openvino_4",
 ]
-"""
-    "hsuwill000/Fluently-v4-LCM-openvino_5",  
-    "hsuwill000/Fluently-v4-LCM-openvino_6",
-    "hsuwill000/Fluently-v4-LCM-openvino_7",
-    "hsuwill000/Fluently-v4-LCM-openvino_8",
-    "hsuwill000/Fluently-v4-LCM-openvino_9",
-    "hsuwill000/Fluently-v4-LCM-openvino_10",
-"""
 
 class CustomClient(Client):
     def __init__(self, *args, timeout=30, **kwargs):
@@ -75,7 +67,13 @@ async def infer_single_gradio(client, prompt):
         traceback.print_exc()  # Print stack trace for debugging
         return None  # Return nothing if an error occurs
 
-async def infer_gradio(prompt: str):
+async def infer_gradio(prompt: str, output_images: gr.Gallery):
+    # Clear previous images from the gallery
+    output_images.update([])
+
+    # Record the start time
+    start_time = time.time()
+
     # Create a list of tasks (one for each repo)
     tasks = []
     for repo in repos:
@@ -93,6 +91,13 @@ async def infer_gradio(prompt: str):
     
     # Run all tasks concurrently (i.e., generate images from all repos)
     results = await asyncio.gather(*tasks)
+
+    # Calculate the time taken for image generation
+    end_time = time.time()
+    time_taken = end_time - start_time
+    print(f"Time taken to generate the image(s): {time_taken:.2f} seconds")
+
+    # Return the results (images)
     return results  # Return all the images as a list
 
 # Define Gradio Interface
@@ -113,7 +118,7 @@ with gr.Blocks() as demo:
     output_images = gr.Gallery(label="Generated Images", elem_id="gallery", show_label=False)
     
     # Connecting the button click to the image generation function
-    run_button.click(infer_gradio, inputs=prompt_input, outputs=output_images)
+    run_button.click(infer_gradio, inputs=[prompt_input, output_images], outputs=output_images)
 
 # Launch Gradio app
 demo.launch()