Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -60,6 +60,7 @@ model.load_state_dict(state_dict)
|
|
60 |
model = model.to(DEVICE).eval()
|
61 |
|
62 |
import torch
|
|
|
63 |
from controlnet_bria import BriaControlNetModel, BriaMultiControlNetModel
|
64 |
from pipeline_bria_controlnet import BriaControlNetPipeline
|
65 |
import PIL.Image as Image
|
|
|
60 |
model = model.to(DEVICE).eval()
|
61 |
|
62 |
import torch
|
63 |
+
from diffusers.utils import load_image
|
64 |
from controlnet_bria import BriaControlNetModel, BriaMultiControlNetModel
|
65 |
from pipeline_bria_controlnet import BriaControlNetPipeline
|
66 |
import PIL.Image as Image
|