Pie31415 commited on
Commit
ad1af47
β€’
1 Parent(s): 59672e3

updated app

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,10 +1,15 @@
1
  import os, sys
 
2
  import argparse
 
3
  import numpy as np
4
  import torch
5
  import matplotlib.pyplot as plt
 
6
  from PIL import Image
7
 
 
 
8
  sys.path.append("./rome")
9
  from src.utils import args as args_utils
10
  from src.utils.processing import process_black_shape, tensor2image
 
1
  import os, sys
2
+ import subprocess
3
  import argparse
4
+
5
  import numpy as np
6
  import torch
7
  import matplotlib.pyplot as plt
8
+
9
  from PIL import Image
10
 
11
+ subprocess.run(["git submodule update", "--init", "--recursive"])
12
+
13
  sys.path.append("./rome")
14
  from src.utils import args as args_utils
15
  from src.utils.processing import process_black_shape, tensor2image