IamVicky111 commited on
Commit
20afda5
·
verified ·
1 Parent(s): 0bac895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,10 +4,13 @@ from transformers.utils import logging
4
  import os
5
  import gradio as gr
6
  from PIL import Image
 
 
 
7
 
8
 
9
 
10
- od_pipe = pipeline("object-detection", "./models/facebook/detr-resnet-50")
11
 
12
  def get_pipeline_prediction(pil_image):
13
 
 
4
  import os
5
  import gradio as gr
6
  from PIL import Image
7
+ # Use a pipeline as a high-level helper
8
+ from transformers import pipeline
9
+
10
 
11
 
12
 
13
+ od_pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
14
 
15
  def get_pipeline_prediction(pil_image):
16