Spaces:

juancamval
/

test

Sleeping

juancamval commited on Apr 12

Commit

d6532bf

verified ·

1 Parent(s): b4f0f17

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,10 @@
 import streamlit as st
 from transformers import pipeline
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("text-generation", model="google/gemma-3-1b-it")
 st.write(pipe(messages))

+import os
 import streamlit as st
 from transformers import pipeline
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
+pipe = pipeline("text-generation", model="google/gemma-3-1b-it",token=os.getenv("HF_TOKEN"))
 st.write(pipe(messages))