Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
import torch
|
4 |
|
5 |
-
# La ruta al modelo
|
6 |
model_path = "AvaLovelace/LLaMA-ASCII-Art"
|
7 |
|
8 |
# Carga el tokenizador y el modelo preentrenados
|
@@ -39,12 +39,12 @@ def generate_ascii_art(prompt):
|
|
39 |
except Exception as e:
|
40 |
return f"Error al generar el arte ASCII: {e}"
|
41 |
|
42 |
-
# Configura la interfaz de Gradio
|
43 |
demo = gr.Interface(
|
44 |
fn=generate_ascii_art,
|
45 |
inputs=gr.Text(label="Prompt"),
|
46 |
outputs=gr.Text(label="ASCII Art"),
|
47 |
-
title="Generador de ASCII
|
48 |
description="Escribe un prompt de texto para generar arte ASCII con un modelo Llama especializado."
|
49 |
)
|
50 |
|
|
|
2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
3 |
import torch
|
4 |
|
5 |
+
# La ruta al modelo apunta al repositorio que me enviaste.
|
6 |
model_path = "AvaLovelace/LLaMA-ASCII-Art"
|
7 |
|
8 |
# Carga el tokenizador y el modelo preentrenados
|
|
|
39 |
except Exception as e:
|
40 |
return f"Error al generar el arte ASCII: {e}"
|
41 |
|
42 |
+
# Configura la interfaz de Gradio
|
43 |
demo = gr.Interface(
|
44 |
fn=generate_ascii_art,
|
45 |
inputs=gr.Text(label="Prompt"),
|
46 |
outputs=gr.Text(label="ASCII Art"),
|
47 |
+
title="Generador de Arte ASCII para Taraxa Pulse",
|
48 |
description="Escribe un prompt de texto para generar arte ASCII con un modelo Llama especializado."
|
49 |
)
|
50 |
|