stupidog04
/

krenzcolor_chkpt_classifier

Image Classification

Model card Files Files and versions Metrics Training metrics Community

stupidog04 commited on Oct 28, 2022

Commit

8f0d639

•

1 Parent(s): cc45a6f

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -17,7 +17,7 @@ class PreTrainedPipeline():
         self.model = self.pipe.model.from_pretrained(path)
-    def __call__(self, inputs)-> Dict[str, str]:
         """
         Args:
             inputs (:obj:`np.array`):

         self.model = self.pipe.model.from_pretrained(path)
+    def __call__(self, inputs):
         """
         Args:
             inputs (:obj:`np.array`):