Update app.py

app.py

@@ -22,41 +22,51 @@ print(f"Python version: {sys.version}")
 
 # Create mock detectron2 structures to prevent import errors
 print("Setting up mock detectron2 module")
-class MockDetectron2Structures:
-    class Boxes:
-        def __init__(self, *args, **kwargs):
-            pass
+class Boxes:
+    def __init__(self, *args, **kwargs):
+        pass
 
-    class ImageList:
-        def __init__(self, *args, **kwargs):
-            pass
-            
-        @staticmethod
-        def from_tensors(*args, **kwargs):
-            return MockDetectron2Structures.ImageList()
+class ImageList:
+    def __init__(self, *args, **kwargs):
+        pass
+        
+    @staticmethod
+    def from_tensors(*args, **kwargs):
+        return ImageList()
 
-    class Instances:
-        def __init__(self, *args, **kwargs):
-            pass
+class Instances:
+    def __init__(self, *args, **kwargs):
+        pass
 
-    class BitMasks:
-        def __init__(self, *args, **kwargs):
-            pass
-            
-        @staticmethod
-        def from_polygon_masks(*args, **kwargs):
-            return MockDetectron2Structures.BitMasks()
+class BitMasks:
+    def __init__(self, *args, **kwargs):
+        pass
+        
+    @staticmethod
+    def from_polygon_masks(*args, **kwargs):
+        return BitMasks()
 
-    class BoxMode:
-        XYXY_ABS = 0
-        XYWH_ABS = 1
+class BoxMode:
+    XYXY_ABS = 0
+    XYWH_ABS = 1
 
-# Add mock detectron2 to sys.modules
+# Add mock detectron2 to sys.modules as a proper package
 if 'detectron2' not in sys.modules:
     import types
     detectron2_module = types.ModuleType('detectron2')
+    structures_module = types.ModuleType('detectron2.structures')
     sys.modules['detectron2'] = detectron2_module
-    detectron2_module.structures = MockDetectron2Structures
+    sys.modules['detectron2.structures'] = structures_module
+    
+    # Add classes to structures module
+    structures_module.Boxes = Boxes
+    structures_module.ImageList = ImageList
+    structures_module.Instances = Instances
+    structures_module.BitMasks = BitMasks
+    structures_module.BoxMode = BoxMode
+    
+    # Set structures as an attribute of detectron2
+    detectron2_module.structures = structures_module
     print("Mock detectron2 module created")
 
 # Make sure utils directory exists
@@ -438,7 +448,7 @@ description = f"""
 article = "SEEM Demo" + (" (Simplified Interface)" if not model_loaded else "")
 inputs = [
     gr.Image(label="[Stroke] Draw on Image", type="pil"), 
-    gr.inputs.CheckboxGroup(choices=["Stroke", "Example", "Text", "Audio", "Video", "Panoptic"], type="value", label="Interactive Mode"), 
+    gr.CheckboxGroup(choices=["Stroke", "Example", "Text", "Audio", "Video", "Panoptic"], label="Interactive Mode"), 
     gr.Image(label="[Example] Draw on Referring Image", type="pil"), 
     gr.Textbox(label="[Text] Referring Text"), 
     gr.Audio(label="[Audio] Referring Audio", source="microphone", type="filepath"),