iamtejanb commited on
Commit
af5c67d
·
verified ·
1 Parent(s): 7042d4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ def get_pipeline_prediction(pil_image):
13
  pipeline_output)
14
  return processed_image
15
 
16
- od_pipe = pipeline("object-detection", "./models/facebook/detr-resnet-50")
 
17
 
18
  demo = gr.Interface(
19
  fn=get_pipeline_prediction,
 
13
  pipeline_output)
14
  return processed_image
15
 
16
+ #od_pipe = pipeline("object-detection", "./models/facebook/detr-resnet-50")
17
+ od_pipe = pipeline("object-detection", model = "facebook/detr-resnet-50")
18
 
19
  demo = gr.Interface(
20
  fn=get_pipeline_prediction,