imanhamid commited on
Commit
f0d04e7
·
1 Parent(s): 768610f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import PIL
4
  import gradio as gr
5
  import os
6
 
7
- # Load model
8
  class_map = ClassMap(['selected_variant'])
9
  backbone = faster_rcnn.backbones.resnet_fpn.resnet18(pretrained=True)
10
  model = faster_rcnn.model(backbone=backbone, num_classes=len(class_map))
 
4
  import gradio as gr
5
  import os
6
 
7
+ # Load modelw
8
  class_map = ClassMap(['selected_variant'])
9
  backbone = faster_rcnn.backbones.resnet_fpn.resnet18(pretrained=True)
10
  model = faster_rcnn.model(backbone=backbone, num_classes=len(class_map))