Disconnect queue when cancelled

gradio-app · Oct 13, 2022 · 97d74ca · 97d74ca
1 parent 6833dd4
commit 97d74ca
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ui/packages/app/src/api.ts b/ui/packages/app/src/api.ts
@@ -255,7 +255,7 @@ export const fn =
 				);
 				// Cancelled jobs are set to complete
 				if (cancels.length > 0) {
-					cancels.forEach((fn_index) =>
+					cancels.forEach((fn_index) => {
 						loading_status.update(
 							fn_index,
 							"complete",
@@ -264,8 +264,9 @@ export const fn =
 							null,
 							null,
 							null
-						)
-					);
+						);
+						ws_map.get(fn_index)?.close();
+					});
 				}
 			} else {
 				loading_status.update(