examples
Browse files
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=["
|
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,
|