Update model.py

model.py

@@ -98,8 +98,8 @@ def gen_model():
     transforms.ToTensor(),
     transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])])
     device="cpu"
-    state = torch.load('gen.pth.tar', map_location='cpu')
-    state = state['state_dict']
+    # state = torch.load('gen.pth.tar', map_location='cpu')
+    # state = state['state_dict']
     gen=Generator()
     G.load_state_dict(torch.load(f"G101.pth", map_location='cpu'))
     G.eval()