Update app.py

app.py

@@ -201,21 +201,20 @@ def zip_and_upload_images(identifier: str, uploaded_files: List[BytesIO], image_
         return s3_path
 
 def train_model(model_inputs):
-    if cli_args.dry_run:
-        logger.info("Skipping model training since --dry-run is enabled.")
-        logger.info(f"model_inputs: {model_inputs}")
-        return
-
-    if cli_args.train_endpoint_url is None:
-        # Use banana backend
-        api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
-        model_key = "1a3b4ce5-164f-4efb-9f4a-c2ad3a930d0b"
-        st.markdown(str(model_inputs))
-        print(model_inputs)
-        _ = banana.run(api_key, model_key, model_inputs)
-    else:
-        # Send request directly to specified url
-        _ = requests.post(cli_args.train_endpoint_url, json=model_inputs)
+    logger.info("Skipping model training since --dry-run is enabled.")
+    logger.info(f"model_inputs: {model_inputs}")
+    st.markdown(str(model_inputs))
+
+    # if cli_args.train_endpoint_url is None:
+    #     # Use banana backend
+    #     api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
+    #     model_key = "1a3b4ce5-164f-4efb-9f4a-c2ad3a930d0b"
+    #     st.markdown(str(model_inputs))
+    #     print(model_inputs)
+    #     _ = banana.run(api_key, model_key, model_inputs)
+    # else:
+    #     # Send request directly to specified url
+    #     _ = requests.post(cli_args.train_endpoint_url, json=model_inputs)
 
 
 def switch_ux_state(new_state: UxState):