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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,9 +1,16 @@
1
  import streamlit as st
2
- from transformers import pipeline
 
 
 
3
 
4
  # Load the text generation pipeline
5
  generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")
6
 
 
 
 
 
7
  # Streamlit app title and description
8
  st.title("Gemma Text Generation App")
9
  st.write("This app generates text based on the input prompt using the Gemma model.")
 
1
  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")
9
 
10
+
11
+ import streamlit as st
12
+ from transformers import pipeline
13
+
14
  # Streamlit app title and description
15
  st.title("Gemma Text Generation App")
16
  st.write("This app generates text based on the input prompt using the Gemma model.")