Fix exception handling

jahaniam · Jan 31, 2024 · 838719a · 838719a
1 parent 31ce760
commit 838719a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/streamlit_app/app.py b/streamlit_app/app.py
@@ -38,7 +38,10 @@ def predict(img, runtime_sm_client, endpoint_name):
 
 @st.cache_resource
 def get_session():
-    runtime_sm_client = boto3.client("sagemaker-runtime")
+    try:
+        runtime_sm_client = boto3.client("sagemaker-runtime")
+    except:
+        runtime_sm_client = None
     return runtime_sm_client