Spaces:

ayse
/

sentiment-classifier

Runtime error

ayse commited on Apr 19, 2022

Commit

01ca571

•

1 Parent(s): 0832201

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ class Model(torch.nn.Module):
     def __init__(self):
         super(Model).__init__()
         self.config = transformers.DistilBertConfig()
-        self.bert = transformers.AutoModelForSequenceClassification.from_pretrained("https://huggingface.co/ayse/distilbert-english-finetuned/resolve/main/model.pth", config=self.config, map_location=device)
         self.tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
     def forward(self, input_text):

     def __init__(self):
         super(Model).__init__()
         self.config = transformers.DistilBertConfig()
+        self.bert = transformers.AutoModelForSequenceClassification.from_pretrained("https://huggingface.co/ayse/distilbert-english-finetuned/resolve/main/model.pth", config=self.config)
         self.tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
     def forward(self, input_text):