fffiloni commited on
Commit
1e18cd1
1 Parent(s): fc26316

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -212,6 +212,7 @@ def infer(image_in, chosen_model):
212
  if image_in == None :
213
  raise gr.Error("Please provide an image input")
214
 
 
215
  if chosen_model == None :
216
  raise gr.Error("Please pick a model")
217
 
@@ -342,7 +343,8 @@ with gr.Blocks(css=css) as demo:
342
  caption,
343
  retry_btn,
344
  result
345
- ]
 
346
  )
347
 
348
  demo.queue(max_size=16).launch(show_api=False)
 
212
  if image_in == None :
213
  raise gr.Error("Please provide an image input")
214
 
215
+ print(f"CHOSEN MODEL: {chosen_model}")
216
  if chosen_model == None :
217
  raise gr.Error("Please pick a model")
218
 
 
343
  caption,
344
  retry_btn,
345
  result
346
+ ],
347
+ concurrency_limit = 4
348
  )
349
 
350
  demo.queue(max_size=16).launch(show_api=False)