Update app.py

app.py

@@ -12,15 +12,18 @@ LABELS = ['NORMAL', 'TUBERCULOSIS', 'PNEUMONIA', 'COVID19']
 
 def predict_input_image(img):
   try:
-      img = Image.open(BytesIO(base64.b64decode(img))).convert('RGB').resize((128,128))
-      img = np.array(img)
-  except:
-      pass
-  img_4d=img.reshape(-1,128,128,3)/img.max()
-  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)}
+      try:
+          img = Image.open(BytesIO(base64.b64decode(img))).convert('RGB').resize((128,128))
+          img = np.array(img)
+      except:
+          pass
+      img_4d=img.reshape(-1,128,128,3)/img.max()
+      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)}
+  except Exception as e:
+    return {"error": str(e)}
 
 with gr.Blocks(title="Chest X-Ray Disease Classification", css="") as demo:
   with gr.Row():