Update app.py

app.py

@@ -11,12 +11,14 @@ import cv2
 import os
 
 # --- CONFIGURATION ---
-# SECURE SETUP: We read the token from Environment Variables now.
-# Do NOT paste your token here directly.
+# SECURE SETUP: Read token from Environment Variables.
 HF_TOKEN = os.getenv("HF_TOKEN") 
-
 TRIPOSR_MODEL = "stabilityai/TripoSR"
 
+# NEW URL: The old "api-inference" URL is deprecated (Error 410). 
+# We use the new Router URL.
+API_URL = f"https://router.huggingface.co/hf-inference/models/{TRIPOSR_MODEL}"
+
 def photo_to_sketch(image):
     """Instant local sketch – always returns PIL Image"""
     print("-> Starting Sketch Generation...")
@@ -64,15 +66,14 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
     files = {"inputs": ("sketch.png", img_bytes, "image/png")}
     data = {"parameters": json.dumps({"prompt": prompt})}
     
-    # Use the token securely
     headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
     # Retry Loop
     for attempt in range(1, 4):
         try:
-            print(f"-> Attempt {attempt}/3: Sending request to TripoSR...")
+            print(f"-> Attempt {attempt}/3: Sending request to {API_URL}...")
             r = requests.post(
-                f"https://api-inference.huggingface.co/models/{TRIPOSR_MODEL}",
+                API_URL,
                 headers=headers,
                 files=files,
                 data=data,
@@ -93,8 +94,12 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
                 return f"data:model/gltf-binary;base64,{b64}", tmp.name
             
             elif r.status_code == 503:
-                print("-> Model is loading (503). Waiting...")
+                print("-> Model is loading (503). Waiting 20s...")
                 time.sleep(20)
+            elif r.status_code == 410:
+                print("-> Error 410: Endpoint deprecated. Trying fallback...")
+                # Fallback to standard inference if router fails, though router is the fix.
+                time.sleep(5)
             elif r.status_code == 401:
                  raise gr.Error("Authentication Failed. Check your HF_TOKEN secret.")
             else:
@@ -105,7 +110,7 @@ def generate_3d_avatar(sketch_image, height, weight, muscle, gender, breast):
             time.sleep(5)
 
     print("-> All attempts failed.")
-    raise gr.Error("Generation Failed. The AI model is busy or timed out. Please try again in 1 minute.")
+    raise gr.Error("Generation Failed. The model is busy. If this persists, the free API tier for TripoSR might be overloaded.")
 
 # =============== UI ===============
 with gr.Blocks(title="SketchToLife") as demo: