Update app.py

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 from gradio_client import Client
 from PIL import Image
 import os
-import time
 import traceback
 import random
 import time
 
+# Create Client instances for the repositories
 clients = [
-    Client("hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD") ,
+    Client("hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD"),
     Client("HelloSun/LCM_Dreamshaper_v7-int8-ov")
 ]
 
@@ -19,10 +19,8 @@ count = 0
 def infer_gradio(prompt: str):
     global count
     random.seed(time.time())
-    # Create a Client instance to communicate with the Hugging Face space
-	#client = Client(random.choice(repos))
-	client = random.choice(clients)
-
+    # Randomly choose a client
+    client = random.choice(clients)
   
     # Prepare the inputs for the prediction
     inputs = {
@@ -31,11 +29,11 @@ def infer_gradio(prompt: str):
     }
 
     try:
-        # Send the request to the model and receive the image
+        # Send the request to the model and receive the result (image URL or file path)
         result = client.predict(inputs, api_name="/infer")
         
-        # Open the resulting image
-        image = Image.open(result)
+        # Assuming the result is a URL or path, use it to open the image
+        image = Image.open(result)  # If the result is a URL, ensure you download it first
         
         # Create a unique filename to save the image
         filename = f"img_{count:08d}.jpg"
@@ -62,13 +60,13 @@ with gr.Blocks() as demo:
     with gr.Row():  # Use a Row to place the prompt input and the button side by side
         prompt_input = gr.Textbox(
             label="Enter Your Prompt", 
-            show_label = "False",
+            show_label=False,
             placeholder="Type your prompt for image generation here",
             lines=1,  # Set the input to be only one line tall
             interactive=True  # Allow user to interact with the textbox
         )
         
-        # Change the button text to "RUN:" and align it with the prompt input
+        # Change the button text to "RUN" and align it with the prompt input
         run_button = gr.Button("RUN")
     
     # Output image display area
@@ -77,4 +75,4 @@ with gr.Blocks() as demo:
     # Connecting the button click to the image generation function
     run_button.click(infer_gradio, inputs=prompt_input, outputs=output_image)
 
-demo.launch()
+demo.launch()