Update app.py

app.py

@@ -9,13 +9,13 @@ HF_TOKEN = os.getenv('HF')
 if not HF_TOKEN:
     raise ValueError("The HF environment variable is not set. Please set it to your Hugging Face token.")
 
-# Authenticate with Hugging Face
+# Authenticate with Hugging Face and save the token to the Git credentials helper
 login(HF_TOKEN, add_to_git_credential=True)
 
 # Load the model and tokenizer using the Hugging Face token
 model_name = "google/gemma-7b-it"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name, use_auth_token=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name, token=HF_TOKEN)
 
 def generate(text):
     try: