Sabbir772 commited on
Commit
09a3166
·
verified ·
1 Parent(s): 53decb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -4,14 +4,14 @@ import gradio as gr
4
  from peft import PeftModel
5
 
6
  # Load model
7
- model_path = "./banglat5_bn_sy" # path inside Space
8
- tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
9
- model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
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)
17
 
 
4
  from peft import PeftModel
5
 
6
  # Load model
7
+ # model_path = "./banglat5_bn_sy" # path inside Space
 
 
 
 
8
  # tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
9
+ # model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
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)
17