Update app.py

app.py

@@ -1,12 +1,15 @@
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer
 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 = GPT2LMHeadModel.from_pretrained(repo_id)
-    tokenizer = GPT2Tokenizer.from_pretrained(repo_id)
+    model = AutoModelForCausalLM.from_pretrained(
+        repo_id,
+        ignore_mismatched_sizes=True  # Ignorar desajuste de tamaños
+    )
+    tokenizer = AutoTokenizer.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.")