Update app.py

app.py

@@ -9,6 +9,7 @@ from flask import Flask, request, jsonify, redirect, render_template
 from huggingface_hub import HfApi, HfFileSystem
 from PIL import Image
 import uuid
+import requests
 
 app = Flask(__name__)
 
@@ -37,7 +38,11 @@ def generate_filename(data, mime_type=None):
 def save_to_dataset(content, filename):
     fs.mkdir(f"datasets/{DATASET_ID}/images", exist_ok=True)
     fs.write_bytes(f"datasets/{DATASET_ID}/images/{filename}", content)
-    return f"https://huggingface.co/datasets/{DATASET_ID}/resolve/main/images/{filename}"
+    url = f"https://huggingface.co/datasets/{DATASET_ID}/resolve/main/images/{filename}"
+    
+    # 获取重定向后的 URL
+    response = requests.head(url, allow_redirects=True)
+    return response.url
 
 @app.route("/", methods=["GET"])
 def index():
@@ -127,3 +132,4 @@ def upload():
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=True)
+