Spaces:
Build error
Build error
updated app
Browse files
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 |
|