mayankchugh-learning commited on
Commit
1f4f3b7
1 Parent(s): d6edd1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,11 +4,11 @@ from PIL import Image, ImageDraw
4
  # Use a pipeline as a high-level helper
5
  from transformers import pipeline
6
 
7
- # pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
8
 
9
- model_path = "../Models/models--facebook--detr-resnet-50/snapshots/1d5f47bd3bdd2c4bbfa585418ffe6da5028b4c0b"
10
 
11
- object_detector = pipeline("object-detection", model=model_path)
12
 
13
  def draw_bounding_boxes(image, object_detections):
14
  """
 
4
  # Use a pipeline as a high-level helper
5
  from transformers import pipeline
6
 
7
+ object_detector = pipeline("object-detection", model="facebook/detr-resnet-50")
8
 
9
+ # model_path = "../Models/models--facebook--detr-resnet-50/snapshots/1d5f47bd3bdd2c4bbfa585418ffe6da5028b4c0b"
10
 
11
+ # object_detector = pipeline("object-detection", model=model_path)
12
 
13
  def draw_bounding_boxes(image, object_detections):
14
  """