salomonsky commited on
Commit
56eb604
1 Parent(s): 757da8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -63,7 +63,10 @@ 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
- return response.generated_text.strip()
 
 
 
67
  except Exception as e:
68
  return f"Error mejorando el prompt: {e}"
69
 
@@ -93,7 +96,7 @@ with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
93
  width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
94
  height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
95
  scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
96
- steps = gr.Slider(label="Pasos", minimum=1, maximum=100, step=1, value=20)
97
  seed = gr.Number(label="Semilla", value=-1)
98
 
99
  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 = ""
67
+ for r in response:
68
+ improved_text += r["generated_text"]
69
+ return improved_text.strip()
70
  except Exception as e:
71
  return f"Error mejorando el prompt: {e}"
72
 
 
96
  width = gr.Slider(label="Ancho", minimum=512, maximum=1280, step=8, value=1280)
97
  height = gr.Slider(label="Alto", minimum=512, maximum=1280, step=8, value=768)
98
  scales = gr.Slider(label="Escalado", minimum=1, maximum=20, step=1, value=10)
99
+ steps = gr.Slider(label="Pasos", minimum=1, maximum 100, step=1, value=20)
100
  seed = gr.Number(label="Semilla", value=-1)
101
 
102
  btn = gr.Button("Generar")