Update app.py

app.py

@@ -105,11 +105,11 @@ weights = "weights.h5"
 def get_grad(img):
     img = img_pros(img)
     grad_img, y_pred = gen_grad_img_single(weights, img)
-    
-    if y_pred[0] > 0.5: y_pred[0] = "cat"
-    else: y_pred[0] = "dog"
+    pred_class = ""
+    if y_pred[0] > 0.5: pred_class = "cat"
+    else: pred_class = "dog"
         
-    text = "Raw Score: " + str(y_pred) + "\nClassification: " + y_pred[0]
+    text = "Raw Score: " + str(y_pred) + "\nClassification: " + pred_class
     return grad_img, text
 
 demo = gr.Interface(