Yaron Koresh commited on
Commit
540ccae
·
verified ·
1 Parent(s): 82b127b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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():
@@ -88,10 +88,10 @@ def Piper(_do,_dont):
88
  return pipe(
89
  _do,
90
  height=256,
91
- width=384,
92
  negative_prompt=_dont,
93
  num_inference_steps=100,
94
- guidance_scale=3
95
  )
96
 
97
  def infer(prompt_en,prompt2_en):
@@ -128,7 +128,7 @@ footer {
128
  max-width: 15cm !important;
129
  }
130
  .image-container {
131
- aspect-ratio: 384 / 256 !important;
132
  }
133
  .dropdown-arrow {
134
  display: none !important;
 
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():
 
88
  return pipe(
89
  _do,
90
  height=256,
91
+ width=448,
92
  negative_prompt=_dont,
93
  num_inference_steps=100,
94
+ guidance_scale=3.5
95
  )
96
 
97
  def infer(prompt_en,prompt2_en):
 
128
  max-width: 15cm !important;
129
  }
130
  .image-container {
131
+ aspect-ratio: 448 / 256 !important;
132
  }
133
  .dropdown-arrow {
134
  display: none !important;