Update app.py

app.py

@@ -4,9 +4,12 @@ from PIL import Image
 import os
 import time
 import traceback
+import random
 
-# Create a Client instance to communicate with the Hugging Face space
-client = Client("https://huggingface.co/spaces/hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD")
+repos = [
+    "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD",
+    "HelloSun/LCM_Dreamshaper_v7-int8-ov"
+]
 
 # Counter for image filenames to avoid overwriting
 count = 0
@@ -15,6 +18,9 @@ count = 0
 def infer_gradio(prompt: str):
     global count
 
+    # Create a Client instance to communicate with the Hugging Face space
+    client = Client(random.choice(repos))
+    
     # Prepare the inputs for the prediction
     inputs = {
         "prompt": prompt,