Spaces:
Build error
Build error
Upload tool
Browse files
pair_classification_tool.py
CHANGED
@@ -7,6 +7,9 @@ class TextPairClassificationTool(PipelineTool):
|
|
7 |
default_checkpoint = "sgugger/bert-finetuned-mrpc"
|
8 |
pre_processor_class = AutoTokenizer
|
9 |
model_class = AutoModelForSequenceClassification
|
|
|
|
|
|
|
10 |
|
11 |
description = (
|
12 |
"This is a tool that classifies if two texts in English are similar or not using the labels 'equivalent' and "
|
|
|
7 |
default_checkpoint = "sgugger/bert-finetuned-mrpc"
|
8 |
pre_processor_class = AutoTokenizer
|
9 |
model_class = AutoModelForSequenceClassification
|
10 |
+
|
11 |
+
inputs = ["text", "text"]
|
12 |
+
outputs = ["text"]
|
13 |
|
14 |
description = (
|
15 |
"This is a tool that classifies if two texts in English are similar or not using the labels 'equivalent' and "
|