Update app.py

app.py

@@ -1,12 +1,18 @@
-
+import torch
 import sys
 sys.path.append('./ControlNetInpaint/')
 from ultralytics import YOLO
 from PIL import Image
 import gradio as gr
 import numpy as np
+import os
+import cv2
+from PIL import Image
+
+from matplotlib import pyplot as plt
 from diffusers import StableDiffusionInpaintPipeline, ControlNetModel, UniPCMultistepScheduler
 from diffusers.utils import load_image
+
 pipe_sd = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16",
@@ -29,11 +35,8 @@ text_prompt="Transform this image into a work of art by changing its style and c
 import os
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 #pipe.to('cuda')
-import cv2
-from PIL import Image
-import numpy as np
-import torch
-from matplotlib import pyplot as plt
+
+
 def image_enhance(image, mask):
   image = np.array(image)
   mask_image = np.array(mask)