Update app.py

app.py

@@ -90,13 +90,20 @@ class RAGSystem:
             )
             
             # Initialize model and tokenizer
+            # Get HuggingFace token
+            hf_token = os.environ.get('HUGGINGFACE_TOKEN') or os.environ.get('HF_TOKEN')
+            if not hf_token:
+                raise ValueError("No Hugging Face token found. Please set HUGGINGFACE_TOKEN in your environment variables")
+
             self.tokenizer = AutoTokenizer.from_pretrained(
                 self.model_name,
+                token=hf_token,  # Add token here
                 trust_remote_code=True
             )
             
             self.model = AutoModelForCausalLM.from_pretrained(
                 self.model_name,
+                token=hf_token,  # Add token here
                 torch_dtype=torch.float16,
                 trust_remote_code=True,
                 device_map="auto"