Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 from transformers import BartTokenizer, BartForConditionalGeneration
+hf_token = os.getenv("HF_TOKEN")
+        if hf_token is None:
+            raise ValueError("HF_TOKEN environment variable is not set.")
 
 # Load model and tokenizer from Hugging Face hub using the provided model name
 model_name = "iimran/SAM-TheSummariserV2"
-tokenizer = BartTokenizer.from_pretrained(model_name)
-model = BartForConditionalGeneration.from_pretrained(model_name)
+tokenizer = BartTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+model = BartForConditionalGeneration.from_pretrained(model_name, use_auth_token=hf_token)
 
-# Define the summarization function
 def summarize(input_text):
     # Tokenize the input text with truncation
     inputs = tokenizer(input_text, max_length=1024, truncation=True, return_tensors="pt")