fffiloni commited on
Commit
cc1f480
1 Parent(s): 2e4870f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torc
44
  pipe = pipe.to("cuda")
45
 
46
  def diffuse(source_img):
 
47
  prompt = "space crew in a spaceship by Claude Monet"
48
  image = pipe(prompt, source_img).images[0]
49
  print(f"DIFFUSED IMG: {image}")
 
44
  pipe = pipe.to("cuda")
45
 
46
  def diffuse(source_img):
47
+ source_img = Image.open(source_img)
48
  prompt = "space crew in a spaceship by Claude Monet"
49
  image = pipe(prompt, source_img).images[0]
50
  print(f"DIFFUSED IMG: {image}")