Update controlnet_module.py

controlnet_module.py

@@ -1,5 +1,6 @@
 import torch
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+from diffusers import StableDiffusionControlNetPipeline, ControlNetMode
+from torchvision.models.detection import keypointrcnn_resnet50_fpn
 from controlnet_aux import OpenposeDetector
 from PIL import Image
 import random
@@ -8,6 +9,8 @@ import numpy as np
 import gradio as gr
 
 
+
+
 class ControlNetProgressCallback:
     def __init__(self, progress, total_steps):
         self.progress = progress