JavierPenyukaRamen24 commited on
Commit
5fc7bdb
·
verified ·
1 Parent(s): 99414a7

Update app_gradio.py

Browse files
Files changed (1) hide show
  1. app_gradio.py +2 -2
app_gradio.py CHANGED
@@ -6,8 +6,8 @@ import gradio as gr
6
  from torchvision import transforms
7
  from timm.data import resolve_data_config
8
 
9
- CKPT_PATH = os.path.join(os.path.dirname(__file__), res_best.pth)
10
- MODEL_NAME = "ResNet‑50"
11
  NUM_CLASSES = 200
12
  LABELS_PATH = os.path.join(os.path.dirname(__file__), "labels.txt") # labels file to use/create
13
 
 
6
  from torchvision import transforms
7
  from timm.data import resolve_data_config
8
 
9
+ CKPT_PATH = os.path.join(os.path.dirname(__file__),"res_best.pth")
10
+ MODEL_NAME = "resnet50"
11
  NUM_CLASSES = 200
12
  LABELS_PATH = os.path.join(os.path.dirname(__file__), "labels.txt") # labels file to use/create
13