Update app.py

app.py

@@ -1,10 +1,11 @@
 from gradio.outputs import Label
 from icevision.all import *
 import PIL
+import torch
 import gradio as gr
 import os
 
-# Load modelw
+# Load model
 class_map = ClassMap(['selected_variant'])
 backbone = faster_rcnn.backbones.resnet_fpn.resnet18(pretrained=True)
 model = faster_rcnn.model(backbone=backbone, num_classes=len(class_map))