Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,9 @@ import streamlit as st
|
|
6 |
import requests
|
7 |
from io import BytesIO
|
8 |
|
|
|
|
|
|
|
9 |
# Set up Hugging Face model pipeline for text generation
|
10 |
pipe = pipeline("text-generation", model="meta-llama/Llama-Guard-3-8B-INT8")
|
11 |
|
|
|
6 |
import requests
|
7 |
from io import BytesIO
|
8 |
|
9 |
+
# Use HF_TOKEN for Hugging Face model access
|
10 |
+
HF_TOKEN = os.getenv("HF_TOKEN") # Ensure this environment variable is set
|
11 |
+
|
12 |
# Set up Hugging Face model pipeline for text generation
|
13 |
pipe = pipeline("text-generation", model="meta-llama/Llama-Guard-3-8B-INT8")
|
14 |
|