Update app.py

app.py

@@ -4,9 +4,9 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 # Load the tokenizer and model
 @st.cache_resource
 def load_model():
-    model_name = "Shuja007/bart_samsum"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    model_path = "./bart_samsum"  # Local directory where the model is stored
+    tokenizer = AutoTokenizer.from_pretrained(model_path)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_path)
     return tokenizer, model
 
 tokenizer, model = load_model()