TalHach61 commited on
Commit
2fe6472
·
verified ·
1 Parent(s): fa993f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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