Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +5 -1
src/streamlit_app.py
CHANGED
|
@@ -5,6 +5,9 @@ import streamlit as st
|
|
| 5 |
from PIL import Image
|
| 6 |
import tensorflow as tf
|
| 7 |
import cv2
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 10 |
# PAGE CONFIG
|
|
@@ -193,7 +196,8 @@ CATEGORIES = {
|
|
| 193 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 194 |
# MODEL LOADING
|
| 195 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 196 |
-
|
|
|
|
| 197 |
IMG_SIZE = (64, 64) # matches your training config
|
| 198 |
|
| 199 |
@st.cache_resource(show_spinner=False)
|
|
|
|
| 5 |
from PIL import Image
|
| 6 |
import tensorflow as tf
|
| 7 |
import cv2
|
| 8 |
+
from pathlib import Path
|
| 9 |
+
|
| 10 |
+
|
| 11 |
|
| 12 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 13 |
# PAGE CONFIG
|
|
|
|
| 196 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 197 |
# MODEL LOADING
|
| 198 |
# βββββββββββββββββββββββββββββββββββββββββββββ
|
| 199 |
+
BASE_DIR = Path(__file__).resolve().parent
|
| 200 |
+
MODEL_PATH = BASE_DIR / "GermanTraffic-sΔ±fΔ±rdanCNN-OS.keras"
|
| 201 |
IMG_SIZE = (64, 64) # matches your training config
|
| 202 |
|
| 203 |
@st.cache_resource(show_spinner=False)
|