Update arvix.py

arvix.py

@@ -65,7 +65,7 @@ def extract_data(category):
             tools.write_data_to_file(id, 'arxiv.txt')
     else:
         for id in data:
-            if len(sanitized_data) >= 3:
+            if len(sanitized_data) >= 2:
                 break
             if tools.check_data_in_file(id, 'arxiv.txt'):
                 continue
@@ -113,7 +113,7 @@ def extract_arxiv_data():
         for temp_id in temp_id_storage:
             all_ids.append(temp_id)
         random.shuffle(all_ids)
-        if len(all_ids) > 3:
+        if len(all_ids) > 2:
             print(f"Found more than 3 papers for {category}.")
             all_ids = all_ids
         category_data['count'] = len(all_ids)