login

app.py

@@ -1,15 +1,13 @@
+import os
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 import torch
-from huggingface_hub import login
-
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-login()
-
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-270m")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-3-270m")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-270m", token=HF_TOKEN)
+model = AutoModelForCausalLM.from_pretrained("google/gemma-3-270m", token=HF_TOKEN)
 
 
 def predict(message, history):