Yaron Koresh commited on
Commit
d4ae41a
·
verified ·
1 Parent(s): 34d26db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -13,8 +13,8 @@ from diffusers import AutoPipelineForText2Image
13
  from pathos.threading import ThreadPool as Pool
14
 
15
  #model_id = "runwayml/stable-diffusion-v1-5"
16
- model_id = "kandinsky-community/kandinsky-3"
17
- #model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
18
 
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
  if torch.cuda.is_available():
@@ -178,6 +178,9 @@ with gr.Blocks(theme=gr.themes.Soft(),css=css,js=js) as demo:
178
  with gr.Row():
179
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
180
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
 
 
181
 
182
  def _ret(idx,p1,p2):
183
  print(f'Starting {idx}: {p1} {p2}')
 
13
  from pathos.threading import ThreadPool as Pool
14
 
15
  #model_id = "runwayml/stable-diffusion-v1-5"
16
+ #model_id = "kandinsky-community/kandinsky-3"
17
+ model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
18
 
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
  if torch.cuda.is_available():
 
178
  with gr.Row():
179
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
180
  result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
181
+ with gr.Row():
182
+ result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
183
+ result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
184
 
185
  def _ret(idx,p1,p2):
186
  print(f'Starting {idx}: {p1} {p2}')