Update app.py

app.py

@@ -6,13 +6,68 @@
 # Written by Xueyan Zou ([email protected]), Jianwei Yang ([email protected])
 # --------------------------------------------------------
 
-# Install detectron2 before any imports
+# Install dependencies and patch files before any imports
 import os
 import sys
 import subprocess
+
+print("Setting up SEEM environment...")
+
+# Create a custom distributed.py file that doesn't need mpi4py
+os.makedirs('utils', exist_ok=True)
+with open('utils/distributed.py', 'w') as f:
+    f.write("""# Custom distributed.py without mpi4py dependency
+import os
+import torch
+import torch.distributed as dist
+
+class MPI:
+    class COMM_WORLD:
+        @staticmethod
+        def Get_rank():
+            return 0
+        @staticmethod
+        def Get_size():
+            return 1
+
+def init_distributed(opt=None):
+    if opt is not None:
+        opt.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        opt.rank = 0
+        opt.world_size = 1
+        opt.gpu = 0
+        return opt
+    
+    return None
+
+def get_rank():
+    return 0
+
+def get_world_size():
+    return 1
+
+def is_main_process():
+    return True
+
+def synchronize():
+    pass
+
+def all_gather(data):
+    return [data]
+
+def reduce_dict(input_dict, average=True):
+    return input_dict
+""")
+print("Created custom distributed.py")
+
+# Install detectron2
 print("Installing detectron2...")
-subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", "git+https://github.com/MaureenZOU/detectron2-xyz.git"])
-print("Detectron2 installation complete!")
+try:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", "git+https://github.com/MaureenZOU/detectron2-xyz.git"])
+    print("Detectron2 installation complete!")
+except Exception as e:
+    print(f"Error installing detectron2: {e}")
+    sys.exit(1)
 
 # Set Python path to include the repository root
 os.environ["PYTHONPATH"] = os.getcwd()