Upload app.py with huggingface_hub

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 from pathlib import Path
 from PIL import Image, ExifTags
@@ -10,6 +11,7 @@ from huggingface_hub import HfApi
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
 logger = logging.getLogger(__name__)
 
+# Token should be set as a secret in the Space settings
 HF_TOKEN = os.environ.get("HF_TOKEN")
 HF_USERNAME = "latterworks"
 DATASET_NAME = "geo-metadata"
@@ -130,16 +132,16 @@ def process_images(image_files):
             "images": [entry.get("file_name") for entry in metadata_list],
             "metadata": metadata_list
         })
-        dataset.push_to_hub(f"{HF_USERNAME}/{DATASET_NAME}", token=HF_TOKEN)
+        dataset.push_to_hub(f"latterworks/geo-metadata", token=HF_TOKEN)
         api = HfApi()
         api.upload_file(
             path_or_fileobj=output_file,
             path_in_repo="metadata.jsonl",
-            repo_id=f"{HF_USERNAME}/{DATASET_NAME}",
+            repo_id=f"latterworks/geo-metadata",
             repo_type="dataset",
             token=HF_TOKEN
         )
-        return f"Processed {len(metadata_list)} images. Metadata saved to {output_file} and uploaded to {HF_USERNAME}/{DATASET_NAME}", output_file
+        return f"Processed {len(metadata_list)} images. Metadata saved to {output_file} and uploaded to latterworks/geo-metadata", output_file
     return f"Processed {len(metadata_list)} images. Metadata saved to {output_file}", output_file
 
 demo = gr.Interface(
@@ -147,7 +149,7 @@ demo = gr.Interface(
     inputs=gr.Files(label="Upload Images", file_types=["image"]),
     outputs=[gr.Textbox(label="Status"), gr.File(label="Download Metadata")],
     title="Geo-Metadata Extractor",
-    description="Upload images to extract metadata (including GPS) and upload to Hugging Face Hub."
+    description="Upload images to extract metadata (including GPS) and automatically upload to latterworks/geo-metadata on Hugging Face Hub."
 )
 
 if __name__ == "__main__":