Update controlnet_module.py
Browse files- controlnet_module.py +1 -0
controlnet_module.py
CHANGED
|
@@ -9,6 +9,7 @@ import gradio as gr
|
|
| 9 |
import torch.nn.functional as F
|
| 10 |
from transformers import Sam2Model, Sam2Processor
|
| 11 |
from scipy import ndimage
|
|
|
|
| 12 |
|
| 13 |
# === CONTROLNET FORTSCHRITTS-CALLBACK (Für Gradio-UI) ===
|
| 14 |
class ControlNetProgressCallback:
|
|
|
|
| 9 |
import torch.nn.functional as F
|
| 10 |
from transformers import Sam2Model, Sam2Processor
|
| 11 |
from scipy import ndimage
|
| 12 |
+
from skimage import measure, morphology
|
| 13 |
|
| 14 |
# === CONTROLNET FORTSCHRITTS-CALLBACK (Für Gradio-UI) ===
|
| 15 |
class ControlNetProgressCallback:
|