Spaces:

NDugar
/

Horse-2-Zebra-CycleGAN

Runtime error

NDugar commited on Apr 17, 2022

Commit

5ee2f62

•

1 Parent(s): 50e1bb0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ from fastai.learner import load_learner
 model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
 def generate_img(img):
-    _,_,preds = learn.get_preds(dl=[b], with_decoded=True)
     return preds

 model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
 def generate_img(img):
+    _,_,preds = model.get_preds(dl=[b], with_decoded=True)
     return preds