fc63 commited on
Commit
da7c7aa
·
verified ·
1 Parent(s): ef027cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch.nn.functional as F
5
 
6
  # model yükleme
7
  model_name = "fc63/gender_prediction_model_from_text"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
  model.eval()
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
5
 
6
  # model yükleme
7
  model_name = "fc63/gender_prediction_model_from_text"
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
9
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
10
  model.eval()
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")