Update app.py

app.py

@@ -4,7 +4,6 @@ import hashlib
 import mimetypes
 from datetime import datetime
 from io import BytesIO
-import requests
 
 from flask import Flask, request, jsonify, redirect, render_template
 from huggingface_hub import HfApi, HfFileSystem
@@ -39,15 +38,7 @@ def save_to_dataset(content, filename):
     path = f"datasets/{DATASET_ID}/resolve/main/images/{filename}"
     fs.mkdir(f"datasets/{DATASET_ID}/images", exist_ok=True)
     fs.write_bytes(f"datasets/{DATASET_ID}/images/{filename}", content)
-    
-    # 获取正确的 URL
-    repo_id = f"datasets/{DATASET_ID}"
-    file_path = f"images/{filename}"
-    url = hf_api.hf_hub_url(repo_id, filename=file_path)
-    
-    # 获取实际的下载 URL
-    response = requests.head(url, allow_redirects=True)
-    return response.url
+    return f"https://hf-mirror.com/datasets/{DATASET_ID}/resolve/main/images/{filename}"
 
 @app.route("/", methods=["GET"])
 def index():