Update app.py

app.py

@@ -1,39 +1,39 @@
 import gradio as gr
-
 import tensorflow as tf
-import os
 import numpy as np
 
-model = tf.keras.models.load_model('model.hdf5',compile=False)
+# Модельді жүктеу
+model = tf.keras.models.load_model('model.hdf5', compile=False)
 
+# Класстар тізімі
 LABELS = ['NORMAL', 'TUBERCULOSIS', 'PNEUMONIA', 'COVID19']
 
+# Суретті өңдеу және болжау функциясы
 def predict_input_image(img):
-  img_4d=img.reshape(-1,128,128,3)/255.0
-  print(img_4d.min())
-  print(img_4d.max())
-  prediction=model.predict(img_4d)[0]
-  return {LABELS[i]: float(prediction[i]) for i in range(4)}
-
-def k():
-  return gr.update(value=None)
-
-with gr.Blocks(title="Lung Disease Classification", css="") as demo:
-  with gr.Row():
-    textmd = gr.Markdown()
-  with gr.Row():
-    with gr.Column(scale=1, min_width=600):
-      image = gr.inputs.Image(shape=(128,128))
-      with gr.Row():
-        clear_btn = gr.Button("Clear")
-        submit_btn = gr.Button("Submit", elem_id="warningk", variant='primary')
-      '''examples = gr.Examples(examples=["COVID19-0.jpg",
-                                       "NORMAL-0.jpeg",
-                                       "COVID19-1.jpg",
-                                       "PNEUMONIA-0.jpeg"], inputs=image)'''
-    label = gr.outputs.Label(num_top_classes=4)
-    
-    clear_btn.click(k, inputs=[], outputs=image)
-    submit_btn.click(predict_input_image, inputs=image, outputs=label)
-
-demo.launch()
+    img = img.resize((128, 128))
+    img_array = np.array(img).reshape(-1, 128, 128, 3) / 255.0
+    prediction = model.predict(img_array)[0]
+    return {LABELS[i]: float(prediction[i]) for i in range(4)}
+
+# Clear функциясы
+def clear_image():
+    return None
+
+# Интерфейс құру
+with gr.Blocks(title="Lung Disease Classification") as demo:
+    gr.Markdown("## Lung Disease Classification Model\nUpload a chest X-ray to predict disease class.")
+
+    with gr.Row():
+        with gr.Column(scale=1, min_width=600):
+            image = gr.Image(type="pil", shape=(128, 128), label="Upload image")
+            with gr.Row():
+                clear_btn = gr.Button("Clear")
+                submit_btn = gr.Button("Submit", variant='primary')
+        label = gr.Label(num_top_classes=4)
+
+    # Батырмаларға функцияларды бекіту
+    clear_btn.click(fn=clear_image, inputs=[], outputs=image)
+    submit_btn.click(fn=predict_input_image, inputs=image, outputs=label)
+
+# Интерфейсті іске қосу
+demo.launch()