rishab1090 commited on
Commit
4e2e70c
·
verified ·
1 Parent(s): 511c175

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,8 +3,11 @@ from fastapi.responses import JSONResponse
3
  from PIL import Image
4
  import numpy as np
5
  import io
6
- from tensorflow.keras.models import load_model
7
- model = load_model("converted_model.keras")
 
 
 
8
 
9
 
10
  app = FastAPI()
 
3
  from PIL import Image
4
  import numpy as np
5
  import io
6
+ from keras.models import load_model
7
+ import os
8
+
9
+ model_path = os.path.join(os.path.dirname(__file__), "converted_model.keras")
10
+ model = load_model(model_path)
11
 
12
 
13
  app = FastAPI()