Update app.py

app.py

@@ -8,7 +8,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 @torch.inference_mode()
 def inference_gan():
-    generator = torch.jit.load("models/mnist-G-torchscript.pt").to(device)
+    generator = torch.jit.load("mnist-G-torchscript.pt").to(device)
     x = torch.randn(30, 256, device='cuda')
     y = generator(x)
     y = y.view(-1, 1, 28, 28)  # reshape y to have 1 channel
@@ -18,7 +18,7 @@ def inference_gan():
 
 @torch.inference_mode()
 def inference_dcgan():
-    generator = torch.jit.load("models/animefacedataset-G2-torchscript.pt").to(device)
+    generator = torch.jit.load("animefacedataset-G2-torchscript.pt").to(device)
     def denorm(img_tensors):
         stats = (0.5, 0.5, 0.5), (0.5, 0.5, 0.5)
         return img_tensors * stats[1][0] + stats[0][0]