BasToTheMax commited on
Commit
7d44206
·
verified ·
1 Parent(s): 667c735

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import gradio as gr
7
 
8
  def detect_objects(image):
9
  # Load the pre-trained DETR model
10
- processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
11
- model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
12
 
13
  inputs = processor(images=image, return_tensors="pt")
14
  outputs = model(**inputs)
 
7
 
8
  def detect_objects(image):
9
  # Load the pre-trained DETR model
10
+ processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101")
11
+ model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101")
12
 
13
  inputs = processor(images=image, return_tensors="pt")
14
  outputs = model(**inputs)