Spaces:

OmerFarooq
/

Grad_cam

Runtime error

OmerFarooq commited on Jul 20, 2023

Commit

9849d45

1 Parent(s): e79dda6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     if y_pred[0] > 0.5: pred_class = "cat"
     else: pred_class = "dog"
+    text = "Raw Score: " + str(y_pred[0]) + "\nClassification: " + pred_class
     return grad_img, text
 demo = gr.Interface(