Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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 |
-
|
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=
|
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:
|
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;
|