Spaces:

Himanshi
/

Face-Cartoonify-for-Video-Call-Privacy

Runtime error

Himanshi commited on May 25, 2022

Commit

a4f3db1

1 Parent(s): 09be573

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,13 +33,15 @@ model = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cpu",
     progress=False
 )
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
-    size=512, device="cpu",side_by_side=False
 )
 def inference(img, method):

     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
+    device="cuda",
     progress=False
 )
+os.system("nvidia-smi")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
+    size=512, device="cuda",side_by_side=False
 )
 def inference(img, method):