Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import json
 import shutil
 import tempfile
-import uvicorn
 from datetime import datetime
 from flask import Flask, request, jsonify
 from huggingface_hub import login, HfApi
@@ -44,7 +43,7 @@ def upload_image():
             image.save(temp_image_path)
 
             # Create metadata
-            path_in_repo = f"data/{guid}.jpg"
+            path_in_repo = f"images/{guid}.jpg"
             metadata = {
                 "file_name": path_in_repo,
                 "guid": guid,
@@ -70,16 +69,7 @@ def upload_image():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-def start_server(port: int = 7860):
-    uvicorn.run(
-        "app:app",
-        host="0.0.0.0",
-        port=port,
-        workers=1,
-        factory=False,
-        # Use UvicornWorker to handle WSGI app
-        worker_class="uvicorn.workers.UvicornWorker"
-    )
-
-if __name__ == "__main__":
-    start_server()
+if __name__ == '__main__':
+    import os
+    port = int(os.environ.get("PORT", 7860))
+    app.run(host="0.0.0.0", port=port)