Update app.py

app.py

@@ -5,8 +5,9 @@ from PIL import Image
 import matplotlib.pyplot as plt
 import cv2
 from torchvision import transforms
-import timm, torchmetrics
+import timm
 from tqdm import tqdm
+import torch.nn.functional as F 
 
 # Your model and other necessary functions
 # Assuming model, device, final_conv, fc_params, cls_names, and any other needed components are defined elsewhere