MindLabUnimib commited on
Commit
31ab39c
·
1 Parent(s): bd093f8

fix: remove batches

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -39,14 +39,10 @@ def generate(prompts: list[str]) -> tuple[list[str], list[dict[str, float]]]:
39
 
40
  return responses, classifier([text + "[SEP]" + response for text, response in zip(texts, responses)])
41
 
42
- def scores(body: str) -> tuple[list[str], list[dict[str, float]]]:
43
- data = json.loads(body)
44
- return data
45
-
46
 
47
  with gr.Blocks() as demo:
48
  gr.Markdown("Welcome")
49
- gr.api(scores, api_name="score", batch=True, max_batch_size=25)
50
 
51
  demo.queue()
52
  demo.launch()
 
39
 
40
  return responses, classifier([text + "[SEP]" + response for text, response in zip(texts, responses)])
41
 
 
 
 
 
42
 
43
  with gr.Blocks() as demo:
44
  gr.Markdown("Welcome")
45
+ gr.api(generate, api_name="scores", batch=False)
46
 
47
  demo.queue()
48
  demo.launch()