Update app.py

app.py

@@ -10,13 +10,6 @@ model_name = "facebook/bart-base"
 model = AutoModelForSeq2SeqLM.from_pretrained(SAVED_MODEL_PATH).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-dataset = load_dataset("samsum")
-#dataset = load_dataset("samsum", download_mode="force_redownload")
-
-
-train_data = dataset["train"]
-validation_data = dataset["validation"] 
-test_data = dataset["test"]
 
 def summarize(text):
     inputs = tokenizer(f"Summarize dialogue >>\n {text}", return_tensors="pt", max_length=1000, truncation=True, padding="max_length").to(device)
@@ -27,8 +20,8 @@ def summarize(text):
 
 iface = gr.Interface(
     fn=summarize,
-    inputs=gr.inputs.Textbox(lines=10, label="Input Dialogue"),
-    outputs=gr.outputs.Textbox(label="Generated Summary")
+    inputs=gr.Textbox(lines=10, label="Input Dialogue"),
+    outputs=gr.Textbox(label="Generated Summary")
 )
 
 iface.launch()