p1atdev commited on
Commit
742e6a4
1 Parent(s): 2b47e09

fix: add sample images

Browse files
Files changed (3) hide show
  1. app.py +3 -1
  2. images/sample.jpg +0 -0
  3. 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=[["./sample.jpg"], ["./sample2.webp"]],
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