Update app.py
Browse files
app.py
CHANGED
@@ -64,7 +64,7 @@ def process_classification(text, model1, model2, tokenizer1):
|
|
64 |
|
65 |
return f"{round(prediction1, 1)}", f"{round(prediction2, 1)}", f"{round(score, 2)}"
|
66 |
|
67 |
-
def all():
|
68 |
return process_ner(text, pipe_ext), process_ner(text, pipe_ais), process_classification(text, model1, model2, tokenizer1)[0], process_classification(text, model1, model2, tokenizer1)[1], process_classification(text, model1, model2, tokenizer1)[2]
|
69 |
|
70 |
# Define Gradio interface
|
|
|
64 |
|
65 |
return f"{round(prediction1, 1)}", f"{round(prediction2, 1)}", f"{round(score, 2)}"
|
66 |
|
67 |
+
def all(text):
|
68 |
return process_ner(text, pipe_ext), process_ner(text, pipe_ais), process_classification(text, model1, model2, tokenizer1)[0], process_classification(text, model1, model2, tokenizer1)[1], process_classification(text, model1, model2, tokenizer1)[2]
|
69 |
|
70 |
# Define Gradio interface
|