xmrt commited on
Commit
99cb033
1 Parent(s): e789313
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import requests
6
  from PIL import Image
7
  from io import BytesIO
8
  from transformers import OwlViTProcessor, OwlViTForObjectDetection
9
-
10
 
11
  # Use GPU if available
12
  if torch.cuda.is_available():
@@ -66,7 +66,7 @@ upload = gr.Interface(
66
  outputs="image",
67
  title="Zero-Shot Object Detection with OWL-ViT",
68
  description=description,
69
- examples=["./examples/IMGP0178.jpg"],)
70
 
71
  web = gr.Interface(
72
  query_image,
 
6
  from PIL import Image
7
  from io import BytesIO
8
  from transformers import OwlViTProcessor, OwlViTForObjectDetection
9
+ import os
10
 
11
  # Use GPU if available
12
  if torch.cuda.is_available():
 
66
  outputs="image",
67
  title="Zero-Shot Object Detection with OWL-ViT",
68
  description=description,
69
+ examples=[os.path.join("examples","IMGP0178.jpg")])
70
 
71
  web = gr.Interface(
72
  query_image,