Update app_gradio.py
Browse files- 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__),
|
| 10 |
-
MODEL_NAME = "
|
| 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 |
|