Update app.py

app.py

@@ -23,12 +23,12 @@ from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
 
 model = torch.load("h2z-85epoch.pth",map_location=torch.device('cpu'))
-def generate_img(img_path):
+def generate_img(img):
     prediction = model(img, training=False)
     return prediction
 
 
-image = gr.inputs.Image(type="filepath")
+image = gr.inputs.Image(type="numpy")
 op = gr.outputs.Image(type="numpy")
 
 iface = gr.Interface(