Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ from lib.dataloader import ImageMixin
|
|
| 13 |
# ===========================================
|
| 14 |
# 1) パスなど(修正があれば適宜変更)
|
| 15 |
# ===========================================
|
| 16 |
-
test_weight = './weight_epoch-
|
| 17 |
parameter = './parameters.json'
|
| 18 |
|
| 19 |
# ===========================================
|
|
@@ -157,9 +157,9 @@ with gr.Blocks(title="Chest X-ray: AP/PA & Rotation Classification") as demo:
|
|
| 157 |
# サンプルファイルは実際のパスに置き換えてください
|
| 158 |
gr.Examples(
|
| 159 |
examples=[
|
| 160 |
-
'./
|
| 161 |
-
'./
|
| 162 |
-
'./
|
| 163 |
],
|
| 164 |
inputs=input_image
|
| 165 |
)
|
|
|
|
| 13 |
# ===========================================
|
| 14 |
# 1) パスなど(修正があれば適宜変更)
|
| 15 |
# ===========================================
|
| 16 |
+
test_weight = './weight_epoch-011_best.pt'
|
| 17 |
parameter = './parameters.json'
|
| 18 |
|
| 19 |
# ===========================================
|
|
|
|
| 157 |
# サンプルファイルは実際のパスに置き換えてください
|
| 158 |
gr.Examples(
|
| 159 |
examples=[
|
| 160 |
+
'./sample/sample_AP_inverted.png',
|
| 161 |
+
'./sample/sample_PA_right.png',
|
| 162 |
+
'./sample/sample_lateral_upright.png'
|
| 163 |
],
|
| 164 |
inputs=input_image
|
| 165 |
)
|