Update api.py
Browse files
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 |
-
|
|
|
|
|
|
|
| 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")
|