Update app.py

app.py

@@ -2,22 +2,24 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# La ruta al modelo ahora apunta a tu repositorio en Hugging Face Hub
-model_path = "TaraxaPulse/asciiartgenerator" 
+# Define la ruta al modelo de Hugging Face.
+# Usa el nombre del modelo que me proporcionaste.
+model_path = "AvaLovelace/LLaMA-ASCII-Art"
 
-# Carga el tokenizador y el modelo preentrenados
+# Carga el tokenizador y el modelo desde el Hub
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path)
+model.to("cuda") # Asumiendo que tu Space tiene una GPU
 
 def generate_ascii_art(prompt):
     """
-    Genera arte ASCII a partir de un prompt de texto.
+    Genera arte ASCII a partir de un prompt de texto usando un modelo Llama.
     """
     if not prompt:
         return "Por favor, introduce un prompt de texto."
 
     try:
-        inputs = tokenizer(prompt, return_tensors="pt")
+        inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
         outputs = model.generate(
             **inputs, 
             max_length=200, 
@@ -33,12 +35,13 @@ def generate_ascii_art(prompt):
     except Exception as e:
         return f"Error al generar el arte ASCII: {e}"
 
+# Configura la interfaz de Gradio
 demo = gr.Interface(
     fn=generate_ascii_art,
     inputs=gr.Text(label="Prompt"),
     outputs=gr.Text(label="ASCII Art"),
     title="Generador de ASCII Art para Taraxa Pulse",
-    description="Escribe un prompt de texto para generar arte ASCII con tu propio modelo."
+    description="Escribe un prompt para generar arte ASCII con un modelo Llama especializado."
 )
 
 if __name__ == "__main__":