Update app.py

app.py

@@ -59,7 +59,7 @@ def inference(input_img, num_gradcam_images=1, target_layer_number=-1, transpare
             npimg = Image.open('./misclassified/' + files[i])
 
             # Use regex to extract target and predicted classes
-            match = re.search(r'Target_(\w+)_Pred_(\w+)_\d+.jpeg', files[i])
+            match = re.search(r'(\w+)_(\w+).png', files[i])
             target_class = match.group(1)
             predicted_class = match.group(2)