Manjushri commited on
Commit
5bf3881
·
verified ·
1 Parent(s): 7325dc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- 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")
 
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")