Update app.py

app.py

@@ -2,6 +2,7 @@ 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
@@ -69,5 +70,12 @@ def upload_image():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+def start_server(port: int = 7860):
+    uvicorn.run(
+        "app:app",
+        host="0.0.0.0",
+        port=port,
+    )
+
+if __name__ == "__main__":
+    start_server()