gfichetdc commited on
Commit
15e8a72
·
1 Parent(s): 0959301

Add example images, remove webcam, cache predictions

Browse files
app.py CHANGED
@@ -9,11 +9,19 @@ def predict(image):
9
 
10
  demo = gr.Interface(
11
  fn=predict,
12
- inputs=gr.Image(type="pil"),
13
  outputs=gr.Label(num_top_classes=2),
14
- title="Casting Defect Detection (ViT)",
15
- description="Fine-tuned ViT-B/16 classifying casting surface images as defective or ok. 99.5% macro F1.",
16
- examples=[],
 
 
 
 
 
 
 
 
17
  )
18
 
19
  demo.launch()
 
9
 
10
  demo = gr.Interface(
11
  fn=predict,
12
+ inputs=gr.Image(type="pil", sources=["upload"]),
13
  outputs=gr.Label(num_top_classes=2),
14
+ title="Casting Defect Detection",
15
+ description="Click an example below to classify it, or upload your own casting image.",
16
+ examples=[
17
+ ["examples/def_1.jpeg"],
18
+ ["examples/def_2.jpeg"],
19
+ ["examples/def_3.jpeg"],
20
+ ["examples/ok_1.jpeg"],
21
+ ["examples/ok_2.jpeg"],
22
+ ["examples/ok_3.jpeg"],
23
+ ],
24
+ cache_examples=True,
25
  )
26
 
27
  demo.launch()
examples/def_1.jpeg ADDED
examples/def_2.jpeg ADDED
examples/def_3.jpeg ADDED
examples/def_4.jpeg ADDED
examples/def_5.jpeg ADDED
examples/ok_1.jpeg ADDED
examples/ok_2.jpeg ADDED
examples/ok_3.jpeg ADDED
examples/ok_4.jpeg ADDED
examples/ok_5.jpeg ADDED