Update app.py

app.py

@@ -5,7 +5,27 @@ from datetime import datetime
 
 # Hugging Face Token
 HF_TOKEN = os.getenv("HF_TOKEN")
-USERNAME = "openfree"  # 특정 사용자 ID 설정
+
+def get_username_from_token():
+    """Retrieve the username using the Hugging Face token"""
+    if not HF_TOKEN:
+        return None
+
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Accept": "application/json"
+    }
+
+    response = requests.get("https://huggingface.co/api/whoami", headers=headers)
+    if response.status_code == 200:
+        return response.json().get("name")
+    return None
+
+# Retrieve the username
+USERNAME = get_username_from_token()
+
+if not USERNAME:
+    raise ValueError("Error: Could not retrieve username. Check your Hugging Face token.")
 
 def format_timestamp(timestamp):
     if timestamp:
@@ -89,4 +109,4 @@ app = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    app.launch()
+    app.launch()