Update app.py

app.py

@@ -4,9 +4,10 @@ import transformers
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-class Model(transformers.modeling_utils.PreTrainedModel):
+class Model(torcn.nn.Module):
     def __init__(self):
         super(Model).__init__()
+        self.config = transformers.DistilBertConfig()
         self.bert = transformers.AutoModelForSequenceClassification.from_pretrained("ayse/distilbert-english-finetuned", config=self.config, map_location=device)
         self.tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")