Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-from datasets import load_dataset
+#from datasets import load_dataset
 import torch
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-SAVED_MODEL_PATH = 'bart_base_full_finetune_save'
+#SAVED_MODEL_PATH = 'bart_base_full_finetune_save'
+SAVED_MODEL_PATH = './bart_base_full_finetune_save
 model_name = "facebook/bart-base"
 model = AutoModelForSeq2SeqLM.from_pretrained(SAVED_MODEL_PATH).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -17,7 +18,6 @@ def summarize(text):
     summary = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=False) for g in summary_ids]
     return summary[0]
 
-##
 iface = gr.Interface(
     fn=summarize,
     inputs=gr.Textbox(lines=10, label="Input Dialogue"),