Update app.py

app.py

@@ -5,23 +5,20 @@ import os
 import time
 import traceback
 
-# Create Client instances for the repositories
-clients = [
-    Client("hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD"),
-    Client("HelloSun/LCM_Dreamshaper_v7-int8-ov")
+repos = [
+    "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD",
+    "HelloSun/LCM_Dreamshaper_v7-int8-ov"
 ]
 
 # Counter for image filenames to avoid overwriting
 count = 0
-
-# Global counter for selecting clients in order
-client_index = 0
+repo_index = 0  # This will keep track of the current repository
 
 # Gradio Interface Function to handle image generation
 def infer_gradio(prompt: str):
-    global count, client_index
-    # Select the current client based on the client_index
-    client = clients[client_index]
+    global count, repo_index
+    # Create a Client instance to communicate with the Hugging Face space
+    client = Client(repos[repo_index])
     
     # Prepare the inputs for the prediction
     inputs = {
@@ -30,7 +27,7 @@ def infer_gradio(prompt: str):
     }
 
     try:
-        # Send the request to the model and receive the result (image URL or file path)
+        # Send the request to the model and receive the image
         result = client.predict(inputs, api_name="/infer")
         
         # Open the resulting image
@@ -46,10 +43,10 @@ def infer_gradio(prompt: str):
         image.save(filename)
         print(f"Saved image as {filename}")
         
-        # Return the image to be displayed in Gradio
-        # Update the client_index to use the next client in the next call
-        client_index = (client_index + 1) % len(clients)  # Cycle through clients
+        # Increment the repo_index to choose the next repository in the list
+        repo_index = (repo_index + 1) % len(repos)  # Cycle through repos list
         
+        # Return the image to be displayed in Gradio
         return image
     
     except Exception as e:
@@ -64,13 +61,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