testing with tensorflow

AndreaLeylavergne · Feb 29, 2024 · df797b0 · df797b0
1 parent 8e97388
commit df797b0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -1,6 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
-from keras.models import load_model
+import tensorflow as tf
+
 # Create the FastAPI app
 app = FastAPI()