Spaces:
Sleeping
Sleeping
Fix weights filename
Browse files- main.py +1 -1
- pcb_weights.weights.h5 +3 -0
main.py
CHANGED
|
@@ -42,7 +42,7 @@ def build_model():
|
|
| 42 |
@app.on_event("startup")
|
| 43 |
async def load_model():
|
| 44 |
global model, class_names
|
| 45 |
-
weights_path = "pcb_weights.h5"
|
| 46 |
if os.path.exists(weights_path):
|
| 47 |
model = build_model()
|
| 48 |
model.load_weights(weights_path)
|
|
|
|
| 42 |
@app.on_event("startup")
|
| 43 |
async def load_model():
|
| 44 |
global model, class_names
|
| 45 |
+
weights_path = "pcb_weights.weights.h5"
|
| 46 |
if os.path.exists(weights_path):
|
| 47 |
model = build_model()
|
| 48 |
model.load_weights(weights_path)
|
pcb_weights.weights.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d03e16881aed5183976f3406346f2afc7d6d3396381bb974928fbe65763701eb
|
| 3 |
+
size 12646688
|