Upload 2 files

app.py

@@ -2,13 +2,15 @@ import streamlit as st
 import fitz  # PyMuPDF for PDF processing
 import pandas as pd
 from transformers import pipeline
+import os
 
-
+# Get the Hugging Face token from the environment variables
+hf_token = os.getenv("HF_API_TOKEN")
 
 # Load the model (Meta-Llama 3.1 8B)
 @st.cache_resource
 def load_model():
-    model = pipeline("text2text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct", use_auth_token=True)
+    model = pipeline("text2text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct", use_auth_token=hf_token)
     return model
 
 model = load_model()