MedCat
/

table_detection_3

Model card Files Files and versions Community

nguyenp99 commited on Apr 23

Commit

a8acdb5

•

1 Parent(s): 4e461d1

Update config.py

Files changed (1) hide show

config.py +1 -1

config.py CHANGED Viewed

@@ -7,7 +7,7 @@ class Config(f.Pipe):
     def __init__(self) -> None:
         super().__init__(
-            DEVICE = 'cuda:1',
             STANDARD_WIDTH = 1024,
             STANDARD_INTERPOLATION = INTER_CUBIC,

     def __init__(self) -> None:
         super().__init__(
+            DEVICE = 'cuda:0',
             STANDARD_WIDTH = 1024,
             STANDARD_INTERPOLATION = INTER_CUBIC,