aravindbethapudi2017 commited on
Commit
a3248fc
·
verified ·
1 Parent(s): c504997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
  # ✅ Load model & tokenizer
6
  model_name = "microsoft/deberta-v3-base" # Change if needed
7
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=16)
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
 
9
 
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
  model.to(device)
 
5
  # ✅ Load model & tokenizer
6
  model_name = "microsoft/deberta-v3-base" # Change if needed
7
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=16)
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
9
+
10
 
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
  model.to(device)