saann commited on
Commit
2a9fee8
·
1 Parent(s): 866e741

Fix model format and pin TF version

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. main.py +1 -1
  3. pcb_model.keras +3 -0
  4. 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.h5"
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.111.0
2
- uvicorn==0.29.0
3
- python-multipart==0.0.9
4
- Pillow==10.3.0
 
5
  numpy==1.26.4
6
- tensorflow[and-cuda]==2.16.1
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