Asnly commited on
Commit
b438acb
·
verified ·
1 Parent(s): 15cff7c

Deploy Space app files

Browse files
Files changed (1) hide show
  1. app.py +21 -4
app.py CHANGED
@@ -550,7 +550,7 @@ def build_app(instance_df: pd.DataFrame, preview_df: pd.DataFrame) -> gr.Blocks:
550
  return demo
551
 
552
 
553
- def main() -> None:
554
  setup_logging()
555
  logger.info("DATASET_REPO = %r", DATASET_REPO)
556
  logger.info("HF_TOKEN set = %s, length = %d", HF_TOKEN is not None, len(HF_TOKEN) if HF_TOKEN else 0)
@@ -558,9 +558,26 @@ def main() -> None:
558
  preview_df = load_preview_dataframe()
559
  instance_df = build_preview_instance_dataframe(preview_df)
560
  logger.info("Loaded %d preview rows for %d preview instance(s).", len(preview_df), len(instance_df))
561
- app = build_app(instance_df, preview_df)
562
- app.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
563
 
564
 
565
- if __name__ == "__main__":
566
  main()
 
550
  return demo
551
 
552
 
553
+ def create_demo() -> gr.Blocks:
554
  setup_logging()
555
  logger.info("DATASET_REPO = %r", DATASET_REPO)
556
  logger.info("HF_TOKEN set = %s, length = %d", HF_TOKEN is not None, len(HF_TOKEN) if HF_TOKEN else 0)
 
558
  preview_df = load_preview_dataframe()
559
  instance_df = build_preview_instance_dataframe(preview_df)
560
  logger.info("Loaded %d preview rows for %d preview instance(s).", len(preview_df), len(instance_df))
561
+ return build_app(instance_df, preview_df)
562
+
563
+
564
+ def gradio_hot_reload_enabled() -> bool:
565
+ return os.environ.get("GRADIO_HOT_RELOAD", "false").lower() in {"1", "true", "yes"}
566
+
567
+
568
+ def main() -> None:
569
+ launch_kwargs: dict[str, Any] = {}
570
+ server_name = os.environ.get("GRADIO_SERVER_NAME")
571
+ server_port = os.environ.get("GRADIO_SERVER_PORT") or os.environ.get("PORT")
572
+ if server_name:
573
+ launch_kwargs["server_name"] = server_name
574
+ if server_port:
575
+ launch_kwargs["server_port"] = int(server_port)
576
+ demo.launch(**launch_kwargs)
577
+
578
+
579
+ demo = create_demo()
580
 
581
 
582
+ if __name__ == "__main__" and not gradio_hot_reload_enabled():
583
  main()