fix bare except error

gizatechxyz · Mar 14, 2024 · 4a4b156 · 4a4b156
1 parent 058cc12
commit 4a4b156
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/giza_actions/model.py b/giza_actions/model.py
@@ -146,7 +146,8 @@ def _set_session(self, model_id: int):
 
             return ort.InferenceSession(onnx_model)
 
-        except:
+        except Exception as e:
+            print(f"Could not download model: {e}")
             return None
 
     def _download_model(self, model_id: int, output_path: str):