Update app.py

app.py

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