Pie31415 commited on
Commit
b8df5f7
·
1 Parent(s): 687db1a

updated app

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -7,6 +7,11 @@ import torch
7
  import matplotlib.pyplot as plt
8
  from PIL import Image
9
 
 
 
 
 
 
10
  sys.path.append("./rome/")
11
  sys.path.append('./DECA')
12
 
 
7
  import matplotlib.pyplot as plt
8
  from PIL import Image
9
 
10
+ import subprocess, sys
11
+ subprocess.check_call(["pip", "install", "torch==1.6.0+cu101",
12
+ "torchvision==0.7.0+cu101",
13
+ "-f", "https://download.pytorch.org/whl/torch_stable.html"])
14
+
15
  sys.path.append("./rome/")
16
  sys.path.append('./DECA')
17