salomonsky commited on
Commit
2713519
1 Parent(s): d8f32ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -63,11 +63,9 @@ def improve_prompt(prompt):
63
  instruction = "Mejora mi prompt para texto a imagen en inglés con estilo, cinematografía, cámaras, atmósfera e iluminación para la mejor calidad, de máximo 200 palabras."
64
  formatted_prompt = f"{instruction}: {prompt}"
65
  response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
66
-
67
- # Aquí asumimos que la respuesta es un string completo, no una lista de objetos.
68
- improved_text = response # Se espera que 'response' sea una cadena de texto.
69
 
70
- return improved_text.strip()
71
  except Exception as e:
72
  return f"Error mejorando el prompt: {e}"
73
 
@@ -97,7 +95,7 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
97
  width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
98
  height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
99
  scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
100
- steps = gr.Slider(label="Pasos", minimum=1, maximum 100, step=1, value=20)
101
  seed = gr.Number(label="Semilla", value=-1)
102
 
103
  btn = gr.Button("Generar")
 
63
  instruction = "Mejora mi prompt para texto a imagen en inglés con estilo, cinematografía, cámaras, atmósfera e iluminación para la mejor calidad, de máximo 200 palabras."
64
  formatted_prompt = f"{instruction}: {prompt}"
65
  response = llm_client.text_generation(formatted_prompt, max_new_tokens=200)
66
+ improved_text = response.strip()
 
 
67
 
68
+ return improved_text
69
  except Exception as e:
70
  return f"Error mejorando el prompt: {e}"
71
 
 
95
  width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
96
  height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
97
  scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
98
+ steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
99
  seed = gr.Number(label="Semilla", value=-1)
100
 
101
  btn = gr.Button("Generar")