Update app.py

app.py

@@ -2,12 +2,15 @@ import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
 
+# Your Hugging Face token
+hf_token = "YOUR_HUGGING_FACE_TOKEN"
+
 # Update this to your model's Hugging Face path
 model_name = "waqasali1707/Bart-Base-Summarization"
 
 try:
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=hf_token)
 except Exception as e:
     st.error(f"Failed to load model: {e}")