Spaces:

akhaliq
/

AnimeGANv2

Running on T4

Ahsen Khaliq commited on Nov 14, 2021

Commit

a9636e0

•

1 Parent(s): b767bd0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ model2 = torch.hub.load(
     "bryandlee/animegan2-pytorch:main",
     "generator",
     pretrained=True, # or give URL to a pretrained model
-    device="cuda", # or "cpu" if you don't have a GPU
     progress=False
 )

     "bryandlee/animegan2-pytorch:main",
     "generator",
     pretrained=True, # or give URL to a pretrained model
+    device="cuda",
     progress=False
 )