MindLabUnimib commited on
Commit
e562a6b
·
1 Parent(s): c574a7f

fix: add type hint for api endpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def classify_pairs(model, tokenizer, prompts, responses):
57
 
58
 
59
  @spaces.GPU()
60
- def generate(submission: list[dict[str, str]], team_id) -> list[dict[str, str | float]]:
61
  ids = [s["id"] for s in submission]
62
  prompts = [s["prompt"] for s in submission]
63
 
 
57
 
58
 
59
  @spaces.GPU()
60
+ def generate(submission: list[dict[str, str]], team_id: str) -> list[dict[str, str | float]]:
61
  ids = [s["id"] for s in submission]
62
  prompts = [s["prompt"] for s in submission]
63