Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -140,7 +140,7 @@ def infer(
|
|
140 |
print(enhanced_prompt)
|
141 |
print('-- generating image --')
|
142 |
#with torch.no_grad():
|
143 |
-
|
144 |
prompt=enhanced_prompt, # This conversion is fine
|
145 |
negative_prompt=negative_prompt,
|
146 |
guidance_scale=guidance_scale,
|
@@ -148,7 +148,7 @@ def infer(
|
|
148 |
width=width,
|
149 |
height=height,
|
150 |
generator=generator
|
151 |
-
|
152 |
print('-- got image --')
|
153 |
image_path = f"sd35m_{seed}.png"
|
154 |
sd_image.save(image_path,optimize=False,compress_level=0)
|
|
|
140 |
print(enhanced_prompt)
|
141 |
print('-- generating image --')
|
142 |
#with torch.no_grad():
|
143 |
+
sd_image = pipe(
|
144 |
prompt=enhanced_prompt, # This conversion is fine
|
145 |
negative_prompt=negative_prompt,
|
146 |
guidance_scale=guidance_scale,
|
|
|
148 |
width=width,
|
149 |
height=height,
|
150 |
generator=generator
|
151 |
+
).images[0]
|
152 |
print('-- got image --')
|
153 |
image_path = f"sd35m_{seed}.png"
|
154 |
sd_image.save(image_path,optimize=False,compress_level=0)
|