Update app.py

app.py

@@ -64,7 +64,7 @@ def load_model(path:str, file_name: str, device: torch.device) -> nn.Module:
                            f'{file_name}',
                            use_auth_token=TOKEN)
     with open(path, 'rb') as f:
-        model = pickle.load(f)['G_ema']
+        model = torch.load(f)['G_ema']
     model.eval()
     model.to(device)
     with torch.inference_mode():