Tahiro20 commited on
Commit
289f6c9
·
verified ·
1 Parent(s): d88b571

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -1
api.py CHANGED
@@ -5,7 +5,9 @@ import io
5
  import tensorflow as tf
6
 
7
  model = tf.keras.models.load_model('946.h5')
8
- print(model.summary())
 
 
9
  app = FastAPI()
10
 
11
  @app.post("/classify")
 
5
  import tensorflow as tf
6
 
7
  model = tf.keras.models.load_model('946.h5')
8
+ model.save("946_new.keras") # use the new Keras format
9
+ model = tf.keras.models.load_model('946_new.keras')
10
+
11
  app = FastAPI()
12
 
13
  @app.post("/classify")