Spaces:

JunhaoZhuang
/

Cobra

Runtime error

JunhaoZhuang commited on 17 days ago

Commit

785ea27

verified ·

1 Parent(s): 29081b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -368,6 +368,7 @@ def process_multi_images(files):
 @spaces.GPU
 def extract_lines(image):
     global line_model
     src = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
     rows = int(np.ceil(src.shape[0] / 16)) * 16

 @spaces.GPU
 def extract_lines(image):
     global line_model
+    line_model.cuda()
     src = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
     rows = int(np.ceil(src.shape[0] / 16)) * 16