Update app.py

app.py

@@ -37,16 +37,6 @@ _S3_PATH_OUTPUT = "s3://gretel-image-synthetics-use2/data/{identifier}/{image_ty
 # Command-line arguments to control some stuff for easier local testing.
 # Eventually may want to move everything into functions and have a
 # if __name__ == "main" setup instead of everything inline.
-parser = argparse.ArgumentParser()
-parser.add_argument(
-    "--dry-run", action="store_true",
-    help="Skip sending train request to backend server.",
-)
-parser.add_argument(
-    "--train-endpoint-url", default=None,
-    help="URL of backend server to send train request to. If None, use hardcoded banana setup.",
-)
-cli_args = parser.parse_args()
 
 if "key" not in st.session_state:
     st.session_state["key"] = uuid.uuid4().hex
@@ -242,20 +232,11 @@ def verify_captcha(captcha_id, captcha_response):
     return {"error": "Failed to verify captcha"}
 
 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))
-        _ = 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)
+    # Use banana backend
+    api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
+    model_key = "1a3b4ce5-164f-4efb-9f4a-c2ad3a930d0b"
+    #st.markdown(str(model_inputs))
+    _ = banana.run(api_key, model_key, model_inputs)
 
 if st.session_state["email_provided"]:
     user_email_input = st.empty()
@@ -324,7 +305,6 @@ if st.session_state["login"]:
                     }
                     st.session_state["model_inputs"] = {
                         "superhero_file_path": dictionary[img][0],
-                        # Use presigned url since backend does not have credentials
                         "person_file_path": generate_s3_get_url(st.session_state["s3_face_file_path"], expiration_seconds=3600),
                         "superhero_prompt": dictionary[img][1],
                         "num_images": 50,