updating with fixes of hf token

app.py

@@ -41,9 +41,29 @@ def load_pipeline():
 def remote_text_encoder(prompts):
     """Encode prompts using remote text encoder API."""
     try:
-        token = get_token()
+        # Try multiple methods to get the token
+        token = None
+        
+        # Method 1: From huggingface_hub
+        try:
+            token = get_token()
+        except:
+            pass
+        
+        # Method 2: From environment variable (Spaces sets this automatically)
         if not token:
-            raise ValueError("HuggingFace token not found. Please login using 'huggingface-cli login'")
+            token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
+        
+        # Method 3: From Spaces secrets
+        if not token:
+            token = os.environ.get("SPACE_TOKEN")
+        
+        if not token:
+            raise ValueError(
+                "HuggingFace token not found. "
+                "If running on Spaces, make sure your Space has access to gated models. "
+                "If running locally, please login using 'huggingface-cli login'"
+            )
         
         response = requests.post(
             "https://remote-text-encoder-flux-2.huggingface.co/predict",
@@ -59,6 +79,18 @@ def remote_text_encoder(prompts):
         
         device = "cuda" if torch.cuda.is_available() else "cpu"
         return prompt_embeds.to(device)
+    except requests.HTTPError as e:
+        if e.response.status_code == 401:
+            raise Exception(
+                "Authentication failed (401). Your HuggingFace token may not have access to this model. "
+                "Please ensure your token has permission to access FLUX.2 models."
+            )
+        elif e.response.status_code == 403:
+            raise Exception(
+                "Access forbidden (403). You may need to accept the model's license agreement on HuggingFace."
+            )
+        else:
+            raise Exception(f"HTTP error {e.response.status_code}: {str(e)}")
     except Exception as e:
         raise Exception(f"Failed to encode prompt: {str(e)}")