face2inpaint

app.py

@@ -6,15 +6,16 @@ import gradio as gr
 #from torch.fx import symbolic_trace
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
+token = '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'
 
-model1 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='face_paint_512_v1',device=device)
-model2 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='face_paint_512_v2',device=device)
-model3 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='celeba_distill', device=device)
-model4 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='paprika',device=device)
+model1 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='face_paint_512_v1',device=device,token=token)
+model2 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='face_paint_512_v2',device=device,token=token)
+model3 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='celeba_distill', device=device,token=token)
+model4 = torch.hub.load('bryandlee/animegan2-pytorch:main','generator',pretrained='paprika',device=device,token=token)
 
 face2paint = torch.hub.load(
     'bryandlee/animegan2-pytorch:main', 'face2paint', 
-    size=512, device=device,side_by_side=False
+    size=512, device=device,side_by_side=False,token=token
 )
 def inference(img, ver):
     if ver == 'version 1':