Sutirtha commited on
Commit
779eb3b
·
verified ·
1 Parent(s): 5ad24ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,11 +1,16 @@
1
  # app.py
2
 
 
 
 
 
3
  import gradio as gr
4
  import torch
5
  from diffusers import StableDiffusionInpaintPipeline
6
  from realesrgan import RealESRGAN
7
  from PIL import Image
8
 
 
9
  # 1. Initialize the inpainting pipeline on CPU (fp32)
10
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
11
  "runwayml/stable-diffusion-inpainting",
 
1
  # app.py
2
 
3
+ import sys
4
+ import torchvision.transforms.functional as F
5
+ sys.modules['torchvision.transforms.functional_tensor'] = F
6
+
7
  import gradio as gr
8
  import torch
9
  from diffusers import StableDiffusionInpaintPipeline
10
  from realesrgan import RealESRGAN
11
  from PIL import Image
12
 
13
+
14
  # 1. Initialize the inpainting pipeline on CPU (fp32)
15
  pipe = StableDiffusionInpaintPipeline.from_pretrained(
16
  "runwayml/stable-diffusion-inpainting",