Spaces:

Cyril666
/

ContourNet

Build error

Cyril666 commited on Jun 1, 2022

Commit

ca7b6f4

1 Parent(s): 8e2468e

First model version

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,9 +13,9 @@ 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,