Update app.py

app.py

@@ -39,13 +39,17 @@ def predict(req: PredictRequest):
         img = None
         
         if req.image_url:
-            # This "User-Agent" is the ID card that lets us download images
-            headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"}
-            response = requests.get(req.image_url, stream=True, headers=headers)
+            # The "Super-Human" header pack
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
+                "Accept": "image/avif,image/webp,image/apng,image/*,*/*;q=0.8",
+                "Referer": "https://google.com" # This tricks the site into thinking you came from a search
+            }
+            
+            response = requests.get(req.image_url, stream=True, headers=headers, timeout=10)
             
             if response.status_code != 200:
-                # This will tell us EXACTLY why it failed (403, 404, etc.)
-                raise Exception(f"External site returned error: {response.status_code}")
+                raise Exception(f"The image site blocked us with error: {response.status_code}. Try a different URL or Base64 upload.")
                 
             img = Image.open(response.raw).convert("RGB")
         # 2. Handle Base64 Input