add

update_almost_agi.py

@@ -41,7 +41,7 @@ ds = load_dataset("nateraw/parti-prompts")["train"]
 parti_prompt_categories = ds["Category"]
 parti_prompt_challenge = ds["Challenge"]
 
-UPLOAD_ORG = "almost-agi-diff"
+UPLOAD_ORG = "Almost-AGI-Diffusion"
 
 def load_non_solved():
     all_datasets = list_datasets(author=SUBMISSION_ORG)
@@ -49,7 +49,7 @@ def load_non_solved():
     
     all_non_solved_image_ids = []
 
-    for _id in relevant_ids[:5]:
+    for _id in relevant_ids:
         try:
             ds = load_dataset(_id)["train"]
         except:
@@ -73,8 +73,10 @@ def main():
 
         votes_column = upvotes
 
-        ds_to_push.add_column("upvotes", votes_column)
+        ds_to_push = ds_to_push.add_column("upvotes", votes_column)
         sorted_ds = ds_to_push.sort("upvotes", reverse=True)
 
-        import ipdb; ipdb.set_trace()
         sorted_ds.push_to_hub(f"{UPLOAD_ORG}/{name}")
+
+if __name__ == "__main__":
+    main()