Update app.py

app.py

@@ -2,8 +2,11 @@ import streamlit as st
 from transformers import pipeline
 import torch
 import os
+from huggingface_hub import login
 
-pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16, token=os.getenv('HG_TOKEN'))
+
+login(token = os.getenv('HF_TOKEN'))
+pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16)
 
 txt = st.text_area("Test this")