Update app.py

app.py

@@ -1,8 +1,18 @@
+from transformers import GPT2Config, GPT2LMHeadModel
+from transformers import CONFIG_MAPPING, MODEL_MAPPING, pipeline
 import gradio as gr
-from transformers import pipeline
 
-# Cargar el modelo desde Hugging Face Hub
-generator = pipeline("text-generation", model="WolfInk/GPT-1.5-High")
+# Paso 1: Registrar el modelo
+class GPT1_5HighConfig(GPT2Config):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+# Registrar la configuración personalizada en los diccionarios de Transformers
+CONFIG_MAPPING["gpt1_5high"] = GPT1_5HighConfig
+MODEL_MAPPING[GPT1_5HighConfig] = GPT2LMHeadModel  # Usa el modelo adecuado
+
+# Paso 2: Crear el pipeline de generación de texto
+generator = pipeline("text-generation", model="WolfInk/GPT-1.5-High", config="gpt1_5high")
 
 def generate_response(prompt, max_tokens, temperature, top_p, top_k, repetition_penalty):
     output = generator(