Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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)
|