Upload folder using huggingface_hub

app.py

@@ -1,10 +1,26 @@
-from flask import Flask
+from flask import Flask, request, jsonify
+import os
 
 app = Flask(__name__)
 
+# A simple way to get a unique ID for each worker
+# In a real scenario, this would be more robust.
+worker_id = os.environ.get("SPACE_ID", "unknown")
+
 @app.route('/')
 def hello_world():
-    return 'Hello, I am an edge worker!'
+    return f'Hello, I am worker {worker_id}!'
+
+@app.route('/execute', methods=['POST'])
+def execute_task():
+    task_data = request.json
+    if not task_data or 'task_data' not in task_data:
+        return jsonify({"error": "Invalid task data"}), 400
+    
+    # "Process" the data
+    result = f"Hello from worker {worker_id}! I processed your data: {task_data['task_data']}"
+    
+    return jsonify({"result": result})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)