Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -37,10 +37,10 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
37 |
return upscaled
|
38 |
else:
|
39 |
return image
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
|
45 |
if Model == "Anime":
|
46 |
anime = DiffusionPipeline.from_pretrained("circulus/canvers-anime-v3.8.1", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("circulus/canvers-anime-v3.8.1")
|
|
|
37 |
return upscaled
|
38 |
else:
|
39 |
return image
|
40 |
+
else:
|
41 |
+
image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
|
42 |
+
torch.cuda.empty_cache()
|
43 |
+
return image
|
44 |
|
45 |
if Model == "Anime":
|
46 |
anime = DiffusionPipeline.from_pretrained("circulus/canvers-anime-v3.8.1", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("circulus/canvers-anime-v3.8.1")
|