krrishD commited on
Commit
3d0f402
1 Parent(s): 09b621d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -200,15 +200,14 @@ class StableDiffusionInpaintingPipeline(DiffusionPipeline):
200
 
201
  return {"sample": image, "nsfw_content_detected": has_nsfw_concept}
202
 
203
- device = "cuda"
204
- model_path = "CompVis/stable-diffusion-v1-4"
205
 
206
- pipe = StableDiffusionInpaintingPipeline.from_pretrained(
207
  model_path,
208
  revision="fp16",
209
  torch_dtype=torch.float16,
210
- use_auth_token=True
211
- ).to(device)
212
 
213
  import gdown
214
  def download_gdrive_url():
 
200
 
201
  return {"sample": image, "nsfw_content_detected": has_nsfw_concept}
202
 
203
+ device = "cuda"
204
+ model_path = "CompVis/stable-diffusion-v1-4"
205
 
206
+ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
207
  model_path,
208
  revision="fp16",
209
  torch_dtype=torch.float16,
210
+ use_auth_token=True).to(device)
 
211
 
212
  import gdown
213
  def download_gdrive_url():