Allow users to provide their own tokens in case they want to upload the model to under their own account

README.md

@@ -52,11 +52,9 @@ This project provides a Streamlit application that converts Hugging Face models
    ```bash
    # Option 1: Environment variables
    export HF_TOKEN="your_token"
-   export HF_USERNAME="your_username"
 
    # Option 2: Create .streamlit/secrets.toml
    echo 'HF_TOKEN = "your_token"' > .streamlit/secrets.toml
-   echo 'HF_USERNAME = "your_username"' >> .streamlit/secrets.toml
    ```
 
 ## Usage

app.py

@@ -8,7 +8,7 @@ from typing import Optional, Tuple
 from urllib.request import urlopen, urlretrieve
 
 import streamlit as st
-from huggingface_hub import HfApi
+from huggingface_hub import whoami, HfApi
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -30,15 +30,17 @@ class Config:
     @classmethod
     def from_env(cls) -> "Config":
         """Create config from environment variables and secrets."""
-        hf_token = st.secrets.get("HF_TOKEN") or os.getenv("HF_TOKEN", "")
+        system_token = st.secrets.get("HF_TOKEN") or os.getenv("HF_TOKEN", "")
+        user_token = st.session_state.get("user_hf_token", "")
         hf_username = (
-            st.secrets.get("HF_USERNAME")
-            or os.getenv("HF_USERNAME")
-            or os.getenv("SPACE_AUTHOR_NAME", "")
+            whoami(token=user_token)["name"]
+            or os.getenv("SPACE_AUTHOR_NAME")
+            or whoami(token=system_token)["name"]
         )
+        hf_token = user_token or system_token
 
-        if not hf_token or not hf_username:
-            raise ValueError("HF_TOKEN and HF_USERNAME must be set")
+        if not hf_token:
+            raise ValueError("HF_TOKEN must be set")
 
         return cls(hf_token=hf_token, hf_username=hf_username)
 
@@ -162,6 +164,12 @@ def main():
         if not input_model_id:
             return
 
+        st.text_input(
+            f"Optional: Your Hugging Face write token. Leave empty to upload under {config.hf_username}'s account.",
+            type="password",
+            key="user_hf_token",
+        )
+
         model_name = (
             input_model_id.replace(f"{config.hf_base_url}/", "")
             .replace("/", "-")