rishab1090 commited on
Commit
1898bee
·
verified ·
1 Parent(s): c1291fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -16,8 +16,13 @@ import os
16
  import keras
17
 
18
  from keras.models import load_model
 
 
 
 
 
19
 
20
- model = load_model("2.keras") # or "models/fixed_model.h5" if in folder
21
 
22
 
23
 
 
16
  import keras
17
 
18
  from keras.models import load_model
19
+ import io
20
+ from keras.src.saving.saving_lib import load_model # needed for Keras 3 .keras files
21
+
22
+ with open("2.keras", "rb") as f:
23
+ byte_data = f.read()
24
 
25
+ model = load_model(io.BytesIO(byte_data))
26
 
27
 
28