Update main.py

main.py

@@ -49,7 +49,7 @@ parser.add_argument('--trained_model', default='weights/craft_mlt_25k.pth', type
 parser.add_argument('--text_threshold', default=0.7, type=float, help='text confidence threshold')
 parser.add_argument('--low_text', default=0.4, type=float, help='text low-bound score')
 parser.add_argument('--link_threshold', default=0.4, type=float, help='link confidence threshold')
-parser.add_argument('--cuda', default=True, type=str2bool, help='Use cuda for inference')
+parser.add_argument('--cuda', default=False, type=str2bool, help='Use cuda for inference')
 parser.add_argument('--canvas_size', default=1280, type=int, help='image size for inference')
 parser.add_argument('--mag_ratio', default=1.5, type=float, help='image magnification ratio')
 parser.add_argument('--poly', default=False, action='store_true', help='enable polygon type')
@@ -339,16 +339,16 @@ def main(image_path0):
         return original_image
 
         
-if __name__ == '__main__':
+# if __name__ == '__main__':
 
 
 
-    iface = gr.Interface(
-        fn=main,
-        inputs=gr.Image(type="filepath", label="Invoice Image"),
-        outputs=gr.Image(type="pil", label="Masked Invoice Image"),
-        live=True
-    )
+#     iface = gr.Interface(
+#         fn=main,
+#         inputs=gr.Image(type="filepath", label="Invoice Image"),
+#         outputs=gr.Image(type="pil", label="Masked Invoice Image"),
+#         live=True
+#     )
 
-    iface.launch()
+#     iface.launch()