Update app.py

app.py

@@ -5,7 +5,8 @@ import streamlit as st
 from transformers import PegasusTokenizer, PegasusForConditionalGeneration
 
 # Load the model and tokenizer
-model_name = "google/pegasus-xsum"  # Using a different Pegasus model
+model_name =  '.'  # Path to the current directory where files are located
+
 
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name)