added try-catch block in the fastapi app for smooth execution if the file is sent via streamlit file_uploader

app.py

@@ -115,8 +115,14 @@ def _file_upload(image_file: UploadFile = File(...)) -> dict:
         a dict as a response json for the post request
     """
     logging.info(image_file)
-    img_str = image_file.file.read()
-    img_decoded = cv2.imdecode(np.frombuffer(img_str, np.uint8), 0)
+    try:
+        # if the file is sent via post request with open()
+        img_str = image_file.file.read()
+        img_decoded = cv2.imdecode(np.frombuffer(img_str, np.uint8), 0)
+    except:
+        # if the file is sent via post request from streamlit
+        img_decoded = cv2.imdecode(np.frombuffer(image_file.getvalue(), np.uint8), 0)
+
     pred_label_str = get_prediction(img_decoded)
     response_json = {"name": image_file.filename, "prediction": pred_label_str}
     logging.info(response_json)