Update app.py

app.py

@@ -6,18 +6,18 @@ import tensorflow_hub as hub
 from PIL import Image
 
 # Load models
-model_initial = keras.models.load_model(
-    "models/initial_model.h5", custom_objects={'KerasLayer': hub.KerasLayer}
-)
-model_tumor = keras.models.load_model(
-    "models/model_tumor.h5", custom_objects={'KerasLayer': hub.KerasLayer}
-)
-model_stroke = keras.models.load_model(
-    "models/model_stroke.h5", custom_objects={'KerasLayer': hub.KerasLayer}
-)
-model_alzheimer = keras.models.load_model(
-    "models/model_alzheimer.h5", custom_objects={'KerasLayer': hub.KerasLayer}
-)
+#model_initial = keras.models.load_model(
+#    "models/initial_model.h5", custom_objects={'KerasLayer': hub.KerasLayer}
+#)
+#model_tumor = keras.models.load_model(
+#    "models/model_tumor.h5", custom_objects={'KerasLayer': hub.KerasLayer}
+#)
+#model_stroke = keras.models.load_model(
+#    "models/model_stroke.h5", custom_objects={'KerasLayer': hub.KerasLayer}
+#)
+#model_alzheimer = keras.models.load_model(
+#    "models/model_alzheimer.h5", custom_objects={'KerasLayer': hub.KerasLayer}
+#)
 
 class CombinedDiseaseModel(tf.keras.Model):
     def __init__(self, model_initial, model_alzheimer, model_tumor, model_stroke):
@@ -62,12 +62,12 @@ class CombinedDiseaseModel(tf.keras.Model):
 
 
 # Initialize the combined model
-cnn_model = CombinedDiseaseModel(
-    model_initial=model_initial,
-    model_alzheimer=model_alzheimer,
-    model_tumor=model_tumor,
-    model_stroke=model_stroke
-)
+#cnn_model = CombinedDiseaseModel(
+#    model_initial=model_initial,
+#    model_alzheimer=model_alzheimer,
+#    model_tumor=model_tumor,
+#    model_stroke=model_stroke
+#)
 
 
 def process_image(image):
@@ -87,6 +87,8 @@ def gradio_interface(patient_info, query_type, image):
     else:
         return "Please upload an image."
 
+def gradio_interface(patient_info, query_type):
+    return f"Patient Info: {patient_info}\nQuery Type: {query_type}"
 
 # Create Gradio app
 iface = gr.Interface(
@@ -100,11 +102,11 @@ iface = gr.Interface(
         ),
         gr.Textbox(
             label="Query Type"
-        ),
-        gr.Image(
-            type="pil",
-            label="Upload an Image",
-        )
+        )#,
+        #gr.Image(
+        #    type="pil",
+        #    label="Upload an Image",
+        #)
     ],
     outputs=gr.Textbox(label="Response", placeholder="The response will appear here..."),
     title="Medical Diagnosis with MRI",