Spaces:

snehilchatterjee
/

intelAIcorrect

Build error

snehilchatterjee commited on Jun 19, 2024

Commit

00ec9ab

verified ·

1 Parent(s): 2b419d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,7 +106,7 @@ def build_generator():
     return Generator(3, 6, 32)
 model=build_generator().to(device)
-model.load_state_dict(torch.load('./method2(0.960).pt', map_location=torch.device('cpu')))
 def numpify(imgs):

     return Generator(3, 6, 32)
 model=build_generator().to(device)
+model.load_state_dict(torch.load('./generator_weight.pt', map_location=torch.device('cpu')))
 def numpify(imgs):