Update app.py

app.py

@@ -1,12 +1,15 @@
 import gradio as gr
 from PIL import Image
 import cv2
+import os, random
 import numpy as np
 from transformers import pipeline
+import PIL.Image
 from diffusers.utils import load_image
 from accelerate import Accelerator
-import torch, os, random, gc
-from diffusers import StableDiffusionControlNetPipeline, StableDiffusionPipeline, ControlNetModel, UniPCMultistepScheduler
+from diffusers import StableDiffusionPipeline
+import torch
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from controlnet_aux import OpenposeDetector
 
 accelerator = Accelerator(cpu=True)
@@ -143,7 +146,7 @@ def plex(mput, prompt, neg_prompt, stips, modal_id, dula, blip, blop):
     openpose_image.save('./fin.png','PNG')
     apol.append(openpose_image)
     images = [apol[2], apol[1]]
-    imoge = pipe([prompt]*2,images,num_inference_steps=stips,negative_prompt=[neg_prompt]*2,controlnet_conditioning_scale=[blip, blop],height=512,width=512,generator=generator)
+    imoge = pipe([prompt]*2,images,num_inference_steps=stips,generator=generator,negative_prompt=[neg_prompt]*2,controlnet_conditioning_scale=[blip, blop])
     for i, imge in enumerate(imoge["images"]):
         apol.append(imge)
     return apol