Spaces:

gradio
/

animeganv2

Running

aliabd HF staff commited on Sep 6, 2022

Commit

4daf591

•

1 Parent(s): 09c43e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,6 @@ model2 = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cuda",
     progress=False
 )
 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cuda")

     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
     progress=False
 )
 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cuda")