Update app.py

app.py

@@ -11,7 +11,6 @@ def get_x(row):
 def get_y(row):
     return row['Buried ODD']
 
-# Load the models into a dictionary
 models = {
     'Ultrasound': load_learner('ODDUltrasound.pkl'),
     'OCT': load_learner('ODDOCT.pkl'),
@@ -36,9 +35,9 @@ def classify_images(img_ultrasound, img_oct, img_fundus, img_fluorescence):
     
     return f"ODD Detection: {final_decision}"
 
-# Define the Gradio interface
-inputs = [gr.inputs.Image(shape=(192, 192), label=f"{modality} Image") for modality in modality_keys]
-output = gr.outputs.Text(label="Final Decision")
+# Adjusted to use the updated Gradio API
+inputs = [gr.Image(shape=(192, 192), label=f"{modality} Image") for modality in modality_keys]
+output = gr.Text(label="Final Decision")
 
 intf = gr.Interface(fn=classify_images, inputs=inputs, outputs=output,
                     title="ODD Detection from Multiple Imaging Modalities",