Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from PIL import Image
 import torch
+import torchvision
+from torchvision.models import EfficientNet_B7_Weights
 import torchvision.models as models
 import torchvision.transforms as transforms
 import cv2
@@ -9,7 +11,7 @@ import openpyxl
 import os
 
 # Load the pre-trained EfficientNet-B7 model
-model = torchvision.models.efficientnet_b7(pretrained=True)
+model = torchvision.models.efficientnet_b7(weights=EfficientNet_B7_Weights.DEFAULT)
 model.eval()
 
 # Define the transformations to be applied to the input image