Spaces:
Sleeping
Sleeping
Fix model format and pin TF version
Browse files- .gitattributes +1 -0
- main.py +1 -1
- pcb_model.keras +3 -0
- requirements.txt +6 -6
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
*.keras filter=lfs diff=lfs merge=lfs -text
|
main.py
CHANGED
|
@@ -18,7 +18,7 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
| 18 |
templates = Jinja2Templates(directory="templates")
|
| 19 |
|
| 20 |
# ── Load model on startup ─────────────────────────────────────────────────────
|
| 21 |
-
MODEL_PATH = "pcb_model.
|
| 22 |
CLASS_PATH = "class_names.json"
|
| 23 |
IMG_SIZE = (224, 224)
|
| 24 |
|
|
|
|
| 18 |
templates = Jinja2Templates(directory="templates")
|
| 19 |
|
| 20 |
# ── Load model on startup ─────────────────────────────────────────────────────
|
| 21 |
+
MODEL_PATH = "pcb_model.keras"
|
| 22 |
CLASS_PATH = "class_names.json"
|
| 23 |
IMG_SIZE = (224, 224)
|
| 24 |
|
pcb_model.keras
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:77b65139e954461ef127bbd20777b5da7e454303d53fac70949f5199b1703e55
|
| 3 |
+
size 12799104
|
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
-
fastapi==0.
|
| 2 |
-
uvicorn==0.
|
| 3 |
-
python-multipart==0.0.
|
| 4 |
-
|
|
|
|
| 5 |
numpy==1.26.4
|
| 6 |
-
tensorflow
|
| 7 |
-
jinja2==3.1.4
|
|
|
|
| 1 |
+
fastapi==0.104.1
|
| 2 |
+
uvicorn[standard]==0.24.0
|
| 3 |
+
python-multipart==0.0.6
|
| 4 |
+
jinja2==3.1.2
|
| 5 |
+
Pillow==10.1.0
|
| 6 |
numpy==1.26.4
|
| 7 |
+
tensorflow==2.16.1
|
|
|