Update app.py

app.py

@@ -1,12 +1,12 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import gradio as gr
 
 # Cargar modelo y tokenizador desde Hugging Face Hub
 repo_id = "Blandskron/ModelChatBot"  # Nombre del repositorio en Hugging Face Hub
 
 try:
-    model = AutoModelForCausalLM.from_pretrained(repo_id)
-    tokenizer = AutoTokenizer.from_pretrained(repo_id)
+    model = GPT2LMHeadModel.from_pretrained(repo_id)
+    tokenizer = GPT2Tokenizer.from_pretrained(repo_id)
 except Exception as e:
     print(f"Error al cargar el modelo: {e}")
     print("Asegúrate de que el repositorio contiene todos los archivos necesarios.")