sgugger commited on
Commit
66e8bb3
1 Parent(s): 2f460b4

Upload tool

Browse files
Files changed (1) hide show
  1. pair_classification_tool.py +3 -0
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 "