Added cpu as map location

app.py

@@ -29,7 +29,7 @@ class Generator(nn.Module):
         return self.main(input)
 
 path = hf_hub_download('huggan/ArtGAN', 'ArtGAN.pt')
-model = torch.load(path)
+model = torch.load(path, map_location=torch.device('cpu'))
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 def generate(seed):