Update app.py

app.py

@@ -24,8 +24,8 @@ from fastai.learner import load_learner
 
 model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
 def generate_img(img):
-    prediction = model(img, training=False)
-    return prediction
+    _,_,preds = learn.get_preds(dl=[b], with_decoded=True)
+    return preds
 
 
 image = gr.inputs.Image(type="numpy")