Guy2 commited on
Commit
58debd5
1 Parent(s): be6d39a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from PIL import Image
8
  import numpy as np
9
 
10
  image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
11
- # model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-150epoch")
12
- model = DetrForObjectDetection.from_pretrained("taroii/airport-security")
13
  id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
14
  def anylize(url):
15
  image = Image.open(requests.get(url, stream=True).raw)
 
8
  import numpy as np
9
 
10
  image_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
11
+ model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-150epoch")
12
+ # model = DetrForObjectDetection.from_pretrained("taroii/airport-security")
13
  id2label = {0: 'dangerous-items', 1: 'Gun', 2: 'Knife', 3: 'Pliers', 4: 'Scissors', 5: 'Wrench'}
14
  def anylize(url):
15
  image = Image.open(requests.get(url, stream=True).raw)