fffiloni commited on
Commit
c65aada
1 Parent(s): a2cc772

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,12 +4,12 @@ import random
4
  from PIL import Image
5
  import cv2
6
 
7
- from diffusers import StableDiffusionImg2ImgPipeline
8
 
9
 
10
- model_id_or_path = "runwayml/stable-diffusion-v1-5"
11
- pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
12
- pipe = pipe.to("cuda")
13
 
14
  import gradio as gr
15
  from glob import glob
 
4
  from PIL import Image
5
  import cv2
6
 
7
+ #from diffusers import StableDiffusionImg2ImgPipeline
8
 
9
 
10
+ #model_id_or_path = "runwayml/stable-diffusion-v1-5"
11
+ #pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
12
+ #pipe = pipe.to("cuda")
13
 
14
  import gradio as gr
15
  from glob import glob