taroii commited on
Commit
232a013
1 Parent(s): b5b878b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def anylize(img):
9
  image = img
10
 
11
  processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
12
- model = DetrForObjectDetection.from_pretrained("taroii/finetuned-detr-50")
13
 
14
  inputs = processor(images=image, return_tensors="pt")
15
  outputs = model(**inputs)
 
9
  image = img
10
 
11
  processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
12
+ model = DetrForObjectDetection.from_pretrained("Guy2/AirportSec-100epoch")
13
 
14
  inputs = processor(images=image, return_tensors="pt")
15
  outputs = model(**inputs)