add

App_main.py

@@ -11,9 +11,9 @@ register_all_modules()
 # os.system('nvidia-smi')
 # os.system('ls /usr/local')
 # os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117')
-# os.system('pip install -U openmim')
-# os.system('mim install mmcv==2.0.0')
-# os.system('mim install mmengine')
+os.system('pip install -U openmim')
+os.system('mim install mmcv==2.0.0')
+os.system('mim install mmengine')
 
 import gradio as gr
 import torch

README.md

@@ -5,7 +5,7 @@ colorFrom: red
 colorTo: yellow
 sdk: gradio
 sdk_version: 3.35.2
-app_file: app.py
+app_file: App_main.py
 pinned: false
 ---
 

requirements.txt

@@ -1,12 +1,9 @@
 torch>=2.0.0
-mmcv>=2.0.0rc4
 importlib_metadata
 mat4py
 
 torchvision
 lightning==2.0.1
-mmengine
-openmim
 wandb
 redis
 scikit-learn