Use tensorflow v1 API (should be updated, but this works as a bridge)

2026-09-21 13:50:20 +08:00 · 2020-08-12 09:15:16 -04:00
parent dd3c64fa8b
commit 2ca4b8d297
1 changed files with 2 additions and 1 deletions
@@ -57,7 +57,8 @@ class ToolRecommendations():
            # keras is not downloaded because of conditional requirement and Galaxy does not build
            try:
                from keras.models import model_from_json
-                import tensorflow as tf
+                import tensorflow.compat.v1 as tf
+                tf.disable_v2_behavior()
            except Exception:
                trans.response.status = 400
                return False