TaraxaPulse commited on
Commit
96a789c
·
verified ·
1 Parent(s): c5c21de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 ahora apunta a tu repositorio en Hugging Face Hub
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 con el endpoint correcto
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 Art para Taraxa Pulse",
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