Update app.py

app.py

@@ -65,12 +65,14 @@ def predict():
                 db_metadata, db_constructor, db_params,
                 config
             )
-        if predictions.get('status') == 999:
-            return jsonify(predictions)
+        
+        # print(predictions)
+        if type(predictions) == str:
+            if predictions == 'null':
+                return jsonify({'predictions': 'null'})
 
         return jsonify({
             'predictions': predictions
         })
-
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860)