Update app.py

app.py

@@ -201,17 +201,17 @@ def zip_and_upload_images(identifier: str, uploaded_files: List[BytesIO], image_
         return s3_path
 
 def train_model(model_inputs):
-    logger.info("Skipping model training since --dry-run is enabled.")
-    logger.info(f"model_inputs: {model_inputs}")
-    st.markdown(str(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)
+    # Use banana backend
+    api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
+    model_key = "bd2c55f5-84bb-40f9-82fb-196ca68b1c1d"
+    #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)