Docker File Update

server/app.py

@@ -234,4 +234,4 @@ def get_result(result_id):
     return jsonify({"error": "Result not found"}), 404
 
 if __name__ == '__main__':
-    app.run(debug=True, port=5000, threaded=True)
+    app.run(host='0.0.0.0', debug=True, port=5000, threaded=True)

server/requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu121
 Flask
 Flask-Cors
 numpy