Spaces:

akhaliq
/

AnimeGANv2

Running on Zero

Ahsen Khaliq commited on Nov 14, 2021

Commit

810d24d

1 Parent(s): a9636e0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 model2 = torch.hub.load(
     "bryandlee/animegan2-pytorch:main",
     "generator",
-    pretrained=True, # or give URL to a pretrained model
     device="cuda",
     progress=False
 )

 model2 = torch.hub.load(
     "bryandlee/animegan2-pytorch:main",
     "generator",
+    pretrained=True,
     device="cuda",
     progress=False
 )