Update app.py

app.py

@@ -1,12 +1,22 @@
 import gradio as gr
 from ultralytics import YOLO
 import tensorflow
+import tensorflow as tf
+from tensorflow.keras.preprocessing import image
+import numpy as np
 import os
 
+model = tf.keras.models.load_model("fine_tuned_resnet50.h5")
+img_dim = (224, 224)
+
 # returning classifiers output
-def image_classifier(inp):
+def predict(img):
+    img = img.resize(img_dim)
+    img_array = image.img_to_array(img) / 255.0
+    img_array = np.expand_dims(img_array, axis=0)
 
-    return None
+    prediction = model.predict(img_array)
+    return prediction
 
 with gr.Blocks() as demo:
     gr.Markdown("# Lung Cancer Classifier")
@@ -16,5 +26,11 @@ with gr.Blocks() as demo:
             submit_btn = gr.Button("Detect Cancer")
         with gr.Column():
             output_text = gr.Textbox(label="Model Results")
+    
+    submit_btn.click(
+        fn=predict,
+        inputs=[input_image],
+        outputs=[output_text]
+    )
             
 demo.launch()