Update app.py

app.py

@@ -2,7 +2,13 @@ from PIL import Image
 import torch
 import gradio as gr
 
-model = torch.hub.load('bryandlee/animegan2-pytorch', 'generator').eval()
+model = 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, # show progress
+)
 
 face2paint = torch.hub.load(
     'bryandlee/animegan2-pytorch:main', 'face2paint',