Ffftdtd5dtft commited on
Commit
f284516
·
verified ·
1 Parent(s): b8e822a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -28,7 +28,6 @@ def load_model():
28
 
29
  MAX_SEED = torch.iinfo(torch.int32).max
30
 
31
- @gr.Interface()
32
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, num_images=1):
33
  load_model() # Asegurarse de que el modelo esté cargado antes de la inferencia
34
 
@@ -141,12 +140,12 @@ with gr.Blocks(css=css) as demo:
141
  cache_examples="lazy"
142
  )
143
 
144
- gr.on(
145
- triggers=[run_button.click, prompt.submit],
146
- fn = infer,
147
- inputs = [prompt, seed, randomize_seed, width, height, num_inference_steps, num_images],
148
- outputs = [results, seed]
149
  )
150
-
151
- # Crear un enlace público con share=True
152
- demo.launch(share=True)
 
28
 
29
  MAX_SEED = torch.iinfo(torch.int32).max
30
 
 
31
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, num_images=1):
32
  load_model() # Asegurarse de que el modelo esté cargado antes de la inferencia
33
 
 
140
  cache_examples="lazy"
141
  )
142
 
143
+ # Conectar el botón y el campo de texto a la función infer
144
+ run_button.click(
145
+ fn=infer,
146
+ inputs=[prompt, seed, randomize_seed, width, height, num_inference_steps, num_images],
147
+ outputs=[results, seed]
148
  )
149
+
150
+ # Crear un enlace público con share=True
151
+ demo.launch(share=True)