behnamsa
/

persian-reverse-dict

Text Classification

reverse-dictionary

Model card Files Files and versions

behnamsa commited on Mar 10, 2023

Commit

d22d58c

·

1 Parent(s): fab0a50

Fix pipeline sample output

Files changed (1) hide show

pipeline.py +8 -6

pipeline.py CHANGED Viewed

@@ -22,12 +22,14 @@ class PreTrainedPipeline():
         self.model = model
-    def __call__(self, inputs):
-        return [ # Sample output, call the model here TODO
-            {'label': 'POSITIVE', 'score': 0.05},
-            {'label': 'NEGATIVE', 'score': 0.03},
-            {'label': 'معنی', 'score': 0.92},
-            {'label': f'{inputs}', 'score': 0},
         ]
     # def RevDict(sent,flag,model):

         self.model = model
+    def __call__(self, inputs: str):
+        return [
+            [ # Sample output, call the model here TODO
+                {'label': 'POSITIVE', 'score': 0.05},
+                {'label': 'NEGATIVE', 'score': 0.03},
+                {'label': 'معنی', 'score': 0.92},
+                {'label': f'{inputs}', 'score': 0},
+            ]
         ]
     # def RevDict(sent,flag,model):