Update app.py

app.py

@@ -9,7 +9,12 @@ model = tf.keras.models.load_model('model.hdf5')
 LABELS = ['NORMAL', 'TUBERCULOSIS', 'PNEUMONIA', 'COVID19']
 
 def predict_input_image(img):
-  img_4d=img.reshape(-1,128,128,3)/255.0
+  try:
+      img = Image.open(BytesIO(base64.b64decode(img)))
+      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]