amazonaws-sp commited on
Commit
2ec55b8
·
verified ·
1 Parent(s): d100824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import numpy as np
10
  import PIL.Image
11
  import spaces
12
  import torch
13
- from diffusers import AutoencoderKL, DiffusionPipeline
14
 
15
  DESCRIPTION = "# Run any LoRA or SD Model"
16
  if not torch.cuda.is_available():
 
10
  import PIL.Image
11
  import spaces
12
  import torch
13
+ from diffusers import AutoencoderKL, DiffusionPipeline, StableDiffusionImg2ImgPipeline
14
 
15
  DESCRIPTION = "# Run any LoRA or SD Model"
16
  if not torch.cuda.is_available():