Update app.py

app.py

@@ -12,7 +12,7 @@ from torch import autocast
 import cv2
 from matplotlib import pyplot as plt
 from torchvision import transforms
-# from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline
 
 import io
 import logging
@@ -95,7 +95,7 @@ def preprocess_mask(mask):
     mask = torch.from_numpy(mask)
     return mask
     
-def process(init_image, mask):
+def model_process(init_image, mask):
     global model
     '''
     input = request.files
@@ -219,7 +219,8 @@ def predict(dict, prompt=""):
     print(f'liuyz_2_', init_image.convert("RGB"))
     print(f'liuyz_3_', init_image.convert("RGB").resize((512, 512)))
     mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
-    output = process(init_image, mask)
+    # output = model_process(init_image, mask)
+    output = None # output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     
     return output.images[0], gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)