Add: import handeling

app.py

@@ -7,6 +7,8 @@ from PIL import Image
 import cv2
 import numpy as np
 import torch
+from yolov5 import xai_yolov5
+from yolov8 import xai_yolov8s
 
 # Sample images directory
 sample_images = {