Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ from PIL import Image
|
|
| 10 |
import gradio as gr
|
| 11 |
|
| 12 |
|
| 13 |
-
CHECKPOINT_PATH = os.environ.get("CKPT_PATH", "
|
| 14 |
|
| 15 |
|
| 16 |
def get_device() -> torch.device:
|
|
@@ -115,9 +115,9 @@ with gr.Blocks(title="ResNet-50 ImageNet-1k Classifier") as demo:
|
|
| 115 |
)
|
| 116 |
gr.Examples(
|
| 117 |
examples=[
|
| 118 |
-
"
|
| 119 |
-
"
|
| 120 |
-
"
|
| 121 |
],
|
| 122 |
inputs=input_images,
|
| 123 |
label="Example images",
|
|
|
|
| 10 |
import gradio as gr
|
| 11 |
|
| 12 |
|
| 13 |
+
CHECKPOINT_PATH = os.environ.get("CKPT_PATH", "best.pth")
|
| 14 |
|
| 15 |
|
| 16 |
def get_device() -> torch.device:
|
|
|
|
| 115 |
)
|
| 116 |
gr.Examples(
|
| 117 |
examples=[
|
| 118 |
+
"/oldfish.png",
|
| 119 |
+
"tiger-shark.png",
|
| 120 |
+
"toilet-tissue.png",
|
| 121 |
],
|
| 122 |
inputs=input_images,
|
| 123 |
label="Example images",
|