Spaces:

CDAO
/

XAITK-Gradio

Sleeping

barry-ravichandran commited on Jan 17, 2024

Commit

d039f64

1 Parent(s): 86183be

Set cuda to false for detection model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,7 +131,7 @@ def get_detection_model(model_choice):
         blackbox_detector = ResNetFRCNN(
             box_thresh=0.05,
             img_batch_size=1,
-            use_cuda=True
         )
     else:

         blackbox_detector = ResNetFRCNN(
             box_thresh=0.05,
             img_batch_size=1,
+            use_cuda=False
         )
     else: