Update gradio_demo.py
Browse files- gradio_demo.py +3 -3
gradio_demo.py
CHANGED
|
@@ -8,9 +8,9 @@ from model_inference import ModelInference
|
|
| 8 |
|
| 9 |
# Load secrets and config from optional .env file
|
| 10 |
load_dotenv()
|
| 11 |
-
GLOBAL_MODEL = global_model_config23_21ep_timm_resnet50.pt
|
| 12 |
-
CATEGORY_MAP = category_map.json
|
| 13 |
-
CATEG_TO_NAME_MAP = categ_to_name_map.json
|
| 14 |
|
| 15 |
|
| 16 |
# Model prediction function
|
|
|
|
| 8 |
|
| 9 |
# Load secrets and config from optional .env file
|
| 10 |
load_dotenv()
|
| 11 |
+
GLOBAL_MODEL = "global_model_config23_21ep_timm_resnet50.pt"
|
| 12 |
+
CATEGORY_MAP = "category_map.json"
|
| 13 |
+
CATEG_TO_NAME_MAP = "categ_to_name_map.json"
|
| 14 |
|
| 15 |
|
| 16 |
# Model prediction function
|