Update app.py

app.py

@@ -6,6 +6,7 @@ import io
 import torch
 from flask import Flask, request
 from PIL import Image
+from waitress import serve
 
 app = Flask(__name__)
 
@@ -31,4 +32,5 @@ def predict():
 
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)  # debug=True causes Restarting with stat
+    #app.run(host="0.0.0.0", port=7860)  # debug=True causes Restarting with stat
+    serve(app, host="0.0.0.0", port=7860)