fix: properly handle OAuth token in dataset push

app.py

@@ -266,6 +266,8 @@ def create_gradio_interface():
             max_words: int,
             public_checkbox: bool,
             collector: OCRDataCollector | None = None,
+            profile: gr.OAuthProfile | None = None,
+            oauth_token: gr.OAuthToken | None = None,
             *args
         ):
             """Handle submission with clean parameter order"""
@@ -334,7 +336,7 @@ def create_gradio_interface():
                 })
 
                 # Push updates to hub
-                dataset.push_to_hub(repo_id, split="train")
+                dataset.push_to_hub(repo_id, split="train", token=oauth_token.token if oauth_token else None)
                 os.remove(temp_path)
 
             new_text = collector.get_random_text_block(max_words)
@@ -348,7 +350,9 @@ def create_gradio_interface():
                 image_input,       # Uploaded image
                 max_words_slider,  # Max words
                 public_checkbox,   # Public dataset option
-                gr.State(collector)  # Pass the collector instance
+                gr.State(collector),
+                gr.State(None),    # Profile will be filled by Gradio
+                gr.State(None)     # Token will be filled by Gradio
             ],
             outputs=[image_input, text_box]
         )