Manjushri commited on
Commit
a38c0be
1 Parent(s): f71aff0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,9 +4,10 @@ import numpy as np
4
  import imageio
5
  from PIL import Image
6
  import torch
 
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
- pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting")
10
  pipe.to(device)
11
 
12
  def resize(height,img):
 
4
  import imageio
5
  from PIL import Image
6
  import torch
7
+ import modin.pandas as pd
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
+ pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", safety_checker=None)
11
  pipe.to(device)
12
 
13
  def resize(height,img):