Update app.py

app.py

@@ -158,9 +158,9 @@ def upload_img():
 
 @sock.route('/pred')
 def echo(ws):
-    while True:
-        data = ws.receive()
-        ws.send(data)
+    data = ws.receive()
+    ws.send(data)
+    return ""
 
 @socketio.on('message')
 def handle_message(message):