Update app.py

app.py

@@ -5,9 +5,11 @@ import os
 import time
 import traceback
 
+api_key = os.getenv('MY_API_KEY')
+
 repos = [
     "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD",
-    "HelloSun/LCM_Dreamshaper_v7-int8-ov"
+    "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD_0"
 ]
 
 # Counter for image filenames to avoid overwriting
@@ -18,7 +20,7 @@ repo_index = 0  # This will keep track of the current repository
 def infer_gradio(prompt: str):
     global count, repo_index
     # Create a Client instance to communicate with the Hugging Face space
-    client = Client(repos[repo_index])
+    client = Client(repos[repo_index], api_key=api_key)
     
     # Prepare the inputs for the prediction
     inputs = {