Figure size changed

app.py

@@ -17,9 +17,9 @@ from torchvision import transforms
 from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 
-import pathlib
-temp = pathlib.PosixPath
-pathlib.PosixPath = pathlib.WindowsPath
+# import pathlib
+# temp = pathlib.PosixPath
+# pathlib.PosixPath = pathlib.WindowsPath
 
 model = load_learner("models/recognizer_model.pkl")
 
@@ -38,7 +38,7 @@ def xai_visualization(image, image_tensor, targeted_category, model, target_laye
 
   mask = grayscale_cam[0, :]
 
-  plt.figure(figsize=(5,5))
+  plt.figure(figsize=(10,5))
 
   plt.axis('off')