bapfalraz commited on
Commit
0d1693b
·
verified ·
1 Parent(s): f0412c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
  import torch
 
4
 
5
-
6
- pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16, use_auth_token=True)
7
 
8
  txt = st.text_area("Test this")
9
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
  import torch
4
+ import os
5
 
6
+ pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16, token=os.getenv('HG_TOKEN'))
 
7
 
8
  txt = st.text_area("Test this")
9