multimodalart HF staff commited on
Commit
3f6a115
1 Parent(s): f829d89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import numpy as np
3
  import cv2
4
  from PIL import Image
 
5
  from region_control import MultiDiffusion, get_views, preprocess_mask
6
  from sketch_helper import get_high_freq_colors, color_quantization, create_binary_matrix
7
  MAX_COLORS = 12
 
2
  import numpy as np
3
  import cv2
4
  from PIL import Image
5
+ import torch
6
  from region_control import MultiDiffusion, get_views, preprocess_mask
7
  from sketch_helper import get_high_freq_colors, color_quantization, create_binary_matrix
8
  MAX_COLORS = 12