Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = [
|