godsofheaven commited on
Commit
ade401c
·
verified ·
1 Parent(s): 05db46e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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", "runs/best.pth")
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
- "input-examples/goldfish.png",
119
- "input-examples/tiger-shark.png",
120
- "input-examples/toilet-tissue.png",
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",