rttl-ai
/

foody-bert

Text Classification

Inference Endpoints

Model card Files Files and versions Community

rttl commited on Sep 21, 2022

Commit

b827343

•

1 Parent(s): 4d90ca7

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -11,7 +11,7 @@ class PreTrainedPipeline():
         Initialize model
         """
         self.bert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
-        self.model = FoodyBertForSequenceClassification.from_pretrained("rttl-ai/foody-bert")
     #def __call__(self, inputs: str) -> List[float]:
     def __call__(self, inputs: str) -> str:
         """

         Initialize model
         """
         self.bert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
+        self.model = FoodyBertForSequenceClassification.from_pretrained(".")
     #def __call__(self, inputs: str) -> List[float]:
     def __call__(self, inputs: str) -> str:
         """