Update app.py

app.py

@@ -1,9 +1,9 @@
-
 import os
 import numpy as np
 import torch
 import torch.nn as nn
 import gradio as gr
+from dataclasses import dataclass
 from torchvision.models import efficientnet_v2_m, EfficientNet_V2_M_Weights
 from torchvision.ops import nms, box_iou
 import torch.nn.functional as F