Guy2 commited on
Commit
e401490
1 Parent(s): 83f2d2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import supervision as sv
5
  import json
6
 
7
  image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
8
- model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-100epoch")
9
  id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
10
  def anylize(img):
11
  image = img
 
5
  import json
6
 
7
  image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
8
+ model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-150epoch")
9
  id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
10
  def anylize(img):
11
  image = img