Upload folder using huggingface_hub

train/side_by_side.py

@@ -117,6 +117,8 @@ def main():
                         help="Hand-picked clip IDs for the side-by-side")
     parser.add_argument("--out", required=True, help="Output HTML path")
     parser.add_argument("--out-json", default=None, help="Optional JSON dump")
+    parser.add_argument("--push-to-space", default=None,
+                        help="If set, e.g. 'aamrinder/subtext-arena', upload outputs to that Space's docs/ dir")
     args = parser.parse_args()
 
     scenarios = load_scenarios()
@@ -211,6 +213,29 @@ def main():
     print(f"  baseline: {n_baseline_correct}/{len(rows)} correct")
     print(f"  trained:  {n_trained_correct}/{len(rows)} correct")
 
+    # Push to HF Space so artifacts survive the ephemeral container
+    if args.push_to_space:
+        try:
+            from huggingface_hub import HfApi
+            api = HfApi()
+            for local, remote in [
+                (args.out, "docs/side_by_side.html"),
+                (args.out_json, "docs/side_by_side.json") if args.out_json else (None, None),
+            ]:
+                if local and Path(local).exists():
+                    api.upload_file(
+                        path_or_fileobj=local,
+                        path_in_repo=remote,
+                        repo_id=args.push_to_space,
+                        repo_type="space",
+                        commit_message="add side-by-side baseline-vs-trained demo",
+                    )
+                    print(f"[done] {remote} pushed to {args.push_to_space}")
+        except Exception as e:
+            print(f"[error] push_to_space failed: {e}")
+    print("\n[main] side-by-side run finished cleanly.")
+    sys.exit(0)
+
 
 if __name__ == "__main__":
     main()