ehristoforu commited on
Commit
51bd190
·
1 Parent(s): b20c8c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -10,7 +10,7 @@ API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
10
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
11
 
12
 
13
- def query(prompt, is_negative=False, image_style="None style", steps=5, cfg_scale=7, seed=None, num_images=4):
14
  images = []
15
  for _ in range(num_images):
16
  if image_style == "None style":
@@ -196,10 +196,11 @@ with gr.Blocks(css=css) as demo:
196
  """
197
  )
198
  with gr.Group():
199
- gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
200
  with gr.Box():
201
  with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
202
  with gr.Column():
 
203
  text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1, elem_id="prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
204
  negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative prompt", max_lines=1, elem_id="negative-prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
205
  text_button = gr.Button("Generate").style(margin=False, rounded=(False, True, True, False), full_width=False)
 
10
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
11
 
12
 
13
+ def query(prompt, is_negative=False, image_style="None style", steps=8, cfg_scale=7, seed=None, num_images=4):
14
  images = []
15
  for _ in range(num_images):
16
  if image_style == "None style":
 
196
  """
197
  )
198
  with gr.Group():
199
+
200
  with gr.Box():
201
  with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
202
  with gr.Column():
203
+ gallery_output = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
204
  text_prompt = gr.Textbox(show_label=False, placeholder="Enter your prompt", max_lines=1, elem_id="prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
205
  negative_prompt = gr.Textbox(show_label=False, placeholder="Enter a negative prompt", max_lines=1, elem_id="negative-prompt-text-input").style(border=(True, False, True, True), rounded=(True, False, False, True), container=False)
206
  text_button = gr.Button("Generate").style(margin=False, rounded=(False, True, True, False), full_width=False)