ayse commited on
Commit
a2cc22d
1 Parent(s): 34b1477

please work

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
6
 
7
  class Model(transformers.modeling_utils.PreTrainedModel):
8
  def __init__(self):
9
- super().__init__()
10
- self.config = transformers.DistilBertConfig()
11
  self.bert = transformers.AutoModelForSequenceClassification.from_pretrained("ayse/distilbert-english-finetuned", config=self.config, map_location=device)
12
  self.tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
13
 
 
6
 
7
  class Model(transformers.modeling_utils.PreTrainedModel):
8
  def __init__(self):
9
+ super(Model).__init__()
 
10
  self.bert = transformers.AutoModelForSequenceClassification.from_pretrained("ayse/distilbert-english-finetuned", config=self.config, map_location=device)
11
  self.tokenizer = transformers.AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
12