Update app.py

app.py

@@ -5,7 +5,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it")
 model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it")
-
+HF_TOKEN = os.getenv('HF')
+login(HF_TOKEN)
 def generate(text):
     try:
         # Tokenize the input text