vCLR

Running on Zero

allencbzhang commited on Apr 3

Commit

6ab8d85

verified ·

1 Parent(s): cabf51c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,6 +35,7 @@ if __name__ == "__main__":
         metadata_dataset="coco_2017_val",
     )
     def inference(img, confidence):
         img = np.array(img)
         _, results = vis_demo.run_on_image(img, confidence)

         metadata_dataset="coco_2017_val",
     )
+    @spaces.GPU
     def inference(img, confidence):
         img = np.array(img)
         _, results = vis_demo.run_on_image(img, confidence)