Sabbir772 commited on
Commit
39b15a2
·
verified ·
1 Parent(s): f6d0867

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from peft import PeftModel
10
 
11
  model_path = "Sabbir772/BnT5Sy"
12
  tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
13
- model = AutoTokenizer.from_pretrained(model_path)
14
  model.eval()
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  model.to(device)
 
10
 
11
  model_path = "Sabbir772/BnT5Sy"
12
  tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
13
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
14
  model.eval()
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  model.to(device)