Update app.py

app.py

@@ -1,12 +1,17 @@
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
+from peft import PeftModel
 
 # Load model
 model_path = "./banglat5_bn_sy"  # path inside Space
 tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
 
+# model_path = "Sabbir772/BnT5Sy"
+# tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False)
+# model = AutoTokenizer.from_pretrained(model_path)
+# model.eval()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)