Update app.py

app.py

@@ -9,7 +9,7 @@ import uvicorn
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 # Load the model
-model_path = 'doctor_ai_model'
+model_path = 'doctor_ai_model.h5'  # Use the correct path and file format
 model = keras.models.load_model(model_path)
 
 # Create FastAPI app
@@ -23,7 +23,7 @@ class InputData(BaseModel):
 @app.post('/predict')
 async def predict(data: InputData):
     input_array = tf.convert_to_tensor(data.input_data)
-    
+
     # Check input shape
     if input_array.shape[1] != 27:
         return {'error': 'Input data must have shape: (None, 27)'}