Spaces:
Running
Running
fix: add sample images
Browse files- app.py +3 -1
- images/sample.jpg +0 -0
- images/sample2.webp +0 -0
app.py
CHANGED
@@ -15,6 +15,8 @@ from modeling_siglip import SiglipForImageClassification
|
|
15 |
MODEL_NAME = "p1atdev/siglip-tagger-test-3"
|
16 |
PROCESSOR_NAME = "google/siglip-so400m-patch14-384"
|
17 |
|
|
|
|
|
18 |
model = SiglipForImageClassification.from_pretrained(
|
19 |
MODEL_NAME,
|
20 |
token=os.environ["HF_READ_TOKEN"],
|
@@ -98,7 +100,7 @@ Example images by NovelAI and niji・journey.
|
|
98 |
start_btn = gr.Button(value="Start", variant="primary")
|
99 |
|
100 |
gr.Examples(
|
101 |
-
examples=
|
102 |
inputs=[input_img],
|
103 |
cache_examples=False,
|
104 |
)
|
|
|
15 |
MODEL_NAME = "p1atdev/siglip-tagger-test-3"
|
16 |
PROCESSOR_NAME = "google/siglip-so400m-patch14-384"
|
17 |
|
18 |
+
EXAMPLES = [["./images/sample.jpg"], ["./images/sample2.webp"]]
|
19 |
+
|
20 |
model = SiglipForImageClassification.from_pretrained(
|
21 |
MODEL_NAME,
|
22 |
token=os.environ["HF_READ_TOKEN"],
|
|
|
100 |
start_btn = gr.Button(value="Start", variant="primary")
|
101 |
|
102 |
gr.Examples(
|
103 |
+
examples=EXAMPLES,
|
104 |
inputs=[input_img],
|
105 |
cache_examples=False,
|
106 |
)
|
images/sample.jpg
ADDED
images/sample2.webp
ADDED