tahirsher commited on
Commit
de0427f
1 Parent(s): 7bde555

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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