Update app.py

app.py

@@ -60,7 +60,6 @@ def process_csv_files(
     max_tokens,
     temperature,
     top_p,
-    hf_token: gr.OAuthToken,
     progress=gr.Progress()
 ):
     """
@@ -69,10 +68,14 @@ def process_csv_files(
     if not files or len(files) == 0:
         return "Please upload at least one CSV file.", None
     
-    if not hf_token or not hf_token.token:
-        return "Please login with your Hugging Face account.", None
+    # Get HF token from environment (automatically available in HF Spaces with OAuth enabled)
+    import os
+    hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN")
     
-    client = InferenceClient(token=hf_token.token, model="meta-llama/Llama-3.3-70B-Instruct")
+    if not hf_token:
+        return "❌ Error: Hugging Face token not found. Please make sure OAuth is enabled for this Space.", None
+    
+    client = InferenceClient(token=hf_token, model="meta-llama/Llama-3.3-70B-Instruct")
     
     output_files = []
     status_messages = []
@@ -167,11 +170,12 @@ with gr.Blocks(title="Business Category Description Generator") as demo:
     CLIP-ready visual descriptions for each category using AI.
     
     **Instructions:**
-    1. Login with your Hugging Face account (required)
-    2. Upload one or more CSV files
-    3. Specify the column name that contains the category keywords
-    4. Click "Process Files" to generate descriptions
-    5. Download the output CSV files
+    1. Upload one or more CSV files
+    2. Specify the column name that contains the category keywords
+    3. Click "Process Files" to generate descriptions
+    4. Download the output CSV files
+    
+    *Note: This Space requires OAuth to be enabled. Make sure to configure it in Space settings.*
     """)
     
     with gr.Sidebar():
@@ -231,8 +235,7 @@ with gr.Blocks(title="Business Category Description Generator") as demo:
             category_column,
             max_tokens,
             temperature,
-            top_p,
-            gr.OAuthToken()
+            top_p
         ],
         outputs=[status_output, files_output]
     )