tahirsher commited on
Commit
e3d0b8a
·
verified ·
1 Parent(s): fb115f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -9,10 +9,11 @@ import requests
9
  from io import BytesIO
10
 
11
  # Use HF_TOKEN for Hugging Face model access
12
- HF_TOKEN = os.getenv("HF_TOKEN") # Ensure this environment variable is set
 
 
 
13
 
14
- # Set up Hugging Face model pipeline for text generation
15
- pipe = pipeline("text-generation", model="meta-llama/Llama-Guard-3-8B-INT8")
16
 
17
  # List of GitHub PDF URLs
18
  PDF_URLS = [
 
9
  from io import BytesIO
10
 
11
  # Use HF_TOKEN for Hugging Face model access
12
+ #HF_TOKEN = os.getenv("HF_TOKEN") # Ensure this environment variable is set
13
+
14
+ # Replace `HF_TOKEN` in the pipeline instantiation directly with your token string (for testing purposes only).
15
+ pipe = pipeline("text-generation", model="meta-llama/Llama-Guard-3-8B-INT8", use_auth_token="HF_TOKEN")
16
 
 
 
17
 
18
  # List of GitHub PDF URLs
19
  PDF_URLS = [