KishoreK commited on
Commit
66e4d07
·
verified ·
1 Parent(s): e8f3051

updated access_token

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import gradio as gr
2
  from transformers import pipeline
 
3
 
4
- pipe = pipeline("text-generation", model="google/gemma-2-2b-it")
 
5
 
6
  def predict(query):
7
  return pipe(query)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ import os
4
 
5
+ access_token = os.getenv("Token")
6
+ pipe = pipeline("text-generation", model="google/gemma-2-2b-it", token=access_token)
7
 
8
  def predict(query):
9
  return pipe(query)