MindLabUnimib commited on
Commit
194674d
·
verified ·
1 Parent(s): c695fab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def generate(submission: list[dict[str, str]], team_id: str) -> list[dict[str, s
80
  prompts = [s["prompt"] for s in submission]
81
 
82
  messages = [[{"role": "user", "content": prompt}] for prompt in prompts]
83
- outputs = model(messages, do_sample=False, temperature=None, max_new_tokens=512, repetition_penalty=1.1, batch_size=25, padding_side="left")
84
  responses = [output[0]["generated_text"][-1]["content"] for output in outputs]
85
 
86
  predictions = classifier([{"text": p, "text_pair": r} for p, r in zip(prompts, responses)], return_all_scores=True, batch_size=25)
 
80
  prompts = [s["prompt"] for s in submission]
81
 
82
  messages = [[{"role": "user", "content": prompt}] for prompt in prompts]
83
+ outputs = model(messages, do_sample=False, temperature=None, max_new_tokens=512, repetition_penalty=1.1, batch_size=25)
84
  responses = [output[0]["generated_text"][-1]["content"] for output in outputs]
85
 
86
  predictions = classifier([{"text": p, "text_pair": r} for p, r in zip(prompts, responses)], return_all_scores=True, batch_size=25)