robertou2 commited on
Commit
3175701
·
1 Parent(s): 400a94c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import re
7
 
8
  from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
9
  from transformers import AutoTokenizer, AutoModelForSequenceClassification,AdamW
10
- tokenizer = AutoTokenizer.from_pretrained('robertou2/twitter_sexismo-finetuned-exist2021')
11
- model = AutoModelForSequenceClassification.from_pretrained("robertou2/twitter_sexismo-finetuned-exist2021")
12
 
13
  import torch
14
  if torch.cuda.is_available():
 
7
 
8
  from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler
9
  from transformers import AutoTokenizer, AutoModelForSequenceClassification,AdamW
10
+ tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/twitter_sexismo-finetuned-exist2021-metwo')
11
+ model = AutoModelForSequenceClassification.from_pretrained("hackathon-pln-es/twitter_sexismo-finetuned-exist2021-metwo")
12
 
13
  import torch
14
  if torch.cuda.is_available():