Spaces:

Cyril666
/

ContourNet

Build error

Cyril666 commited on Jun 1, 2022

Commit

2155fe1

1 Parent(s): ca7b6f4

First model version

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from maskrcnn_benchmark.config import cfg
 def infer(filepath):
     cfg.merge_from_file('./configs/ctw/r50_baseline.yaml')
     # manual override some options
-    #cfg.merge_from_list(["MODEL.DEVICE", "cpu"])
     text_demo = TextDemo(
         cfg,

 def infer(filepath):
     cfg.merge_from_file('./configs/ctw/r50_baseline.yaml')
     # manual override some options
+    cfg.merge_from_list(["MODEL.DEVICE", "cpu"])
     text_demo = TextDemo(
         cfg,