Update app.py

app.py

@@ -10,7 +10,7 @@ LABELS = ['NORMAL', 'TUBERCULOSIS', 'PNEUMONIA', 'COVID19']
 
 # Суретті өңдеу және болжау функциясы
 def predict_input_image(img):
-    img = img.resize((128, 128))
+    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)}
@@ -25,7 +25,7 @@ with gr.Blocks(title="Lung Disease Classification") as demo:
 
     with gr.Row():
         with gr.Column(scale=1, min_width=600):
-            image = gr.Image(type="pil", shape=(128, 128), label="Upload image")
+            image = gr.Image(type="pil", label="Upload image")
             with gr.Row():
                 clear_btn = gr.Button("Clear")
                 submit_btn = gr.Button("Submit", variant='primary')
@@ -36,4 +36,4 @@ with gr.Blocks(title="Lung Disease Classification") as demo:
     submit_btn.click(fn=predict_input_image, inputs=image, outputs=label)
 
 # Интерфейсті іске қосу
-demo.launch()
+demo.launch()