hsuwill000 commited on
Commit
13a2046
·
verified ·
1 Parent(s): c248483

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -7,9 +7,9 @@ import traceback
7
  import random
8
  import time
9
 
10
- repos = [
11
- "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD",
12
- "HelloSun/LCM_Dreamshaper_v7-int8-ov"
13
  ]
14
 
15
  # Counter for image filenames to avoid overwriting
@@ -20,7 +20,9 @@ def infer_gradio(prompt: str):
20
  global count
21
  random.seed(time.time())
22
  # Create a Client instance to communicate with the Hugging Face space
23
- client = Client(random.choice(repos))
 
 
24
 
25
  # Prepare the inputs for the prediction
26
  inputs = {
@@ -75,4 +77,4 @@ with gr.Blocks() as demo:
75
  # Connecting the button click to the image generation function
76
  run_button.click(infer_gradio, inputs=prompt_input, outputs=output_image)
77
 
78
- demo.launch()
 
7
  import random
8
  import time
9
 
10
+ clients = [
11
+ Client("hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD") ,
12
+ Client("HelloSun/LCM_Dreamshaper_v7-int8-ov")
13
  ]
14
 
15
  # Counter for image filenames to avoid overwriting
 
20
  global count
21
  random.seed(time.time())
22
  # Create a Client instance to communicate with the Hugging Face space
23
+ #client = Client(random.choice(repos))
24
+ client = random.choice(clients)
25
+
26
 
27
  # Prepare the inputs for the prediction
28
  inputs = {
 
77
  # Connecting the button click to the image generation function
78
  run_button.click(infer_gradio, inputs=prompt_input, outputs=output_image)
79
 
80
+ demo.launch()