Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -8,9 +8,11 @@ import os
|
|
8 |
import datetime
|
9 |
from PIL import Image
|
10 |
from remove_bg import RemoveBackground
|
|
|
11 |
|
12 |
net_C = hf_hub_download("radames/PIFu-upright-standing", filename="net_C")
|
13 |
net_G = hf_hub_download("radames/PIFu-upright-standing", filename="net_G")
|
|
|
14 |
|
15 |
remove_bg = RemoveBackground()
|
16 |
|
|
|
8 |
import datetime
|
9 |
from PIL import Image
|
10 |
from remove_bg import RemoveBackground
|
11 |
+
import torch
|
12 |
|
13 |
net_C = hf_hub_download("radames/PIFu-upright-standing", filename="net_C")
|
14 |
net_G = hf_hub_download("radames/PIFu-upright-standing", filename="net_G")
|
15 |
+
torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretrained=True)
|
16 |
|
17 |
remove_bg = RemoveBackground()
|
18 |
|