Update app.py

app.py

@@ -21,6 +21,7 @@ backbone = model_type.backbones.resnet50_fpn
 #model = model_type.model(backbone=backbone(pretrained=True), num_classes=len(parser.class_map), **extra_args) 
 
 import models
+from models import model.pth
 
 
 # Transforms
@@ -49,6 +50,7 @@ article="<p style='text-align: center'><a href='https://dicksonneoh.com/fridge-d
 enable_queue=True
 #examples = [['sample_images/3.jpg']]
 examples = [["sample_images/"+file] for file in files] 
+
 def show_preds(input_image, display_label, display_bbox, detection_threshold):
     if detection_threshold==0: detection_threshold=0.5
     img = PIL.Image.fromarray(input_image, 'RGB')