Update app.py

app.py

@@ -3,8 +3,11 @@ import tensorflow as tf
 from tensorflow import keras
 import gradio as gr
 
+# Suppress TensorFlow logs
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+
 # Load the model
-model_path = 'doctor_ai_model.h5'  # or .h5
+model_path = 'doctor_ai_model.h5'  # Change to your model file
 model = keras.models.load_model(model_path)
 
 def predict(input_data):
@@ -25,10 +28,11 @@ def predict(input_data):
     return predicted_class
 
 # Create Gradio interface
-iface = gr.Interface(fn=predict, 
-                     inputs=gr.inputs.Dataframe(type='numpy', 
-                                                 label='Input Data (should have shape (None, 27))'), 
-                     outputs='json')
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Dataframe(type='numpy', label='Input Data (should have shape (None, 27))'),
+    outputs='json'
+)
 
 # Launch the Gradio app
 iface.launch()