Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
|
14 |
|
15 |
def prediction(header, text):
|
16 |
results = classifier(header + " [SEP] " + text)
|
17 |
-
return results[0]["label"],
|
18 |
|
19 |
|
20 |
gradio_ui = gr.Interface(
|
|
|
14 |
|
15 |
def prediction(header, text):
|
16 |
results = classifier(header + " [SEP] " + text)
|
17 |
+
return results[0]["label"], results[0]["score"]
|
18 |
|
19 |
|
20 |
gradio_ui = gr.Interface(
|