Update app.py

app.py

@@ -1,10 +1,16 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 import gradio as gr
-from huggingface_hub import notebook_login
+import os
 
-# Llama a esta función y sigue las instrucciones, ingresando tu token cuando se te solicite.
-notebook_login()
+# Asegúrate de que tu token de Hugging Face está cargado como una variable de entorno
+hf_token = os.environ.get("token")
+
+if hf_token is not None:
+    from huggingface_hub import HfFolder
+    HfFolder.save_token(hf_token)
+else:
+    print("No se encontró el token de Hugging Face. Asegúrate de que la variable de entorno HF_TOKEN esté configurada.")
 
 # Cargar el tokenizador
 tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")