Slfagrouche commited on
Commit
c68af39
·
verified ·
1 Parent(s): 87476e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import streamlit as st
2
  from transformers import pipeline, set_access_token
3
 
4
  # Set up the Hugging Face API token
5
- set_access_token(hugging_face_api_token)
 
6
 
7
  # Load the text generation pipeline
8
  generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")
 
2
  from transformers import pipeline, set_access_token
3
 
4
  # Set up the Hugging Face API token
5
+
6
+ os.environ["HF_HOME"] = hugging_face_api_token
7
 
8
  # Load the text generation pipeline
9
  generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")