Spaces:
Sleeping
Sleeping
Commit
·
82dae9a
1
Parent(s):
3039ab0
feat: format classifier prompt
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def generate(message):
|
|
| 37 |
]
|
| 38 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 39 |
|
| 40 |
-
return response, classifier(text)
|
| 41 |
|
| 42 |
|
| 43 |
demo = gr.Interface(fn=generate, inputs=gr.Text(), outputs=gr.Text())
|
|
|
|
| 37 |
]
|
| 38 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 39 |
|
| 40 |
+
return response, classifier(text + "[SEP]" + response)
|
| 41 |
|
| 42 |
|
| 43 |
demo = gr.Interface(fn=generate, inputs=gr.Text(), outputs=gr.Text())
|