adding keras_model

AndreaLeylavergne · Mar 1, 2024 · 4dc1a82 · 4dc1a82
1 parent 8c60a23
commit 4dc1a82
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -2,6 +2,7 @@
 from pydantic import BaseModel
 from contextlib import asynccontextmanager
 import pickle
+from keras.models import load_model
 
 # Create the FastAPI app
 app = FastAPI()