MindLabUnimib commited on
Commit
1a241d5
·
1 Parent(s): ff8bdc6

feat: add batch param

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,6 +39,6 @@ def generate(prompts: list[str]) -> tuple[list[str], list[dict[str, float]]]:
39
 
40
  with gr.Blocks() as demo:
41
  gr.Markdown("Welcome")
42
- gr.api(generate, api_name="predict")
43
 
44
  demo.launch()
 
39
 
40
  with gr.Blocks() as demo:
41
  gr.Markdown("Welcome")
42
+ gr.api(generate, api_name="predict", batch=True)
43
 
44
  demo.launch()