Update app.py

app.py

@@ -94,8 +94,12 @@ def gen_grad_img_single(weights, img, alpha = 0.4):
 weights = "weights.h5"
 # img, y_pred = gen_grad_img_single(weights, img)
 
+def get_grad(img):
+    a = gen_grad_img_single(weights, img)
+    return a
+
 demo = gr.Interface(
-  fn = gen_grad_img_single,
+  fn = get_grad,
   inputs = gr.Image(type = "pil", shape = (224,224)),
   # outputs = [gr.outputs.Label(num_top_classes = 2, label = 'Classifiaction'), gr.Textbox('infer_time', label = 'Inference Time(ms)')]
   outputs = [gr.Image(type = "numpy"), gr.Textbox('y_pred', label = 'Prediction')]