Update app.py

app.py

@@ -8,9 +8,11 @@ import os
 import datetime
 from PIL import Image
 from remove_bg import RemoveBackground
+import torch
 
 net_C = hf_hub_download("radames/PIFu-upright-standing", filename="net_C")
 net_G = hf_hub_download("radames/PIFu-upright-standing", filename="net_G")
+torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretrained=True)
 
 remove_bg = RemoveBackground()