JackIsNotInTheBox commited on
Commit
40ba3a7
·
1 Parent(s): ff7f3e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -1,4 +1,15 @@
1
  import os
 
 
 
 
 
 
 
 
 
 
 
2
  import torch
3
  import numpy as np
4
  import random
 
1
  import os
2
+ import subprocess
3
+ import sys
4
+
5
+ try:
6
+ import mmcv
7
+ print("mmcv already installed")
8
+ except ImportError:
9
+ print("Installing mmcv via mim...")
10
+ subprocess.check_call([sys.executable, "-m", "mim", "install", "mmcv>=2.0.0"])
11
+ print("mmcv installed successfully")
12
+
13
  import torch
14
  import numpy as np
15
  import random