From 11af1d933890faad6e88173eae538a61954cbdc4 Mon Sep 17 00:00:00 2001
From: Hafiidz <3688500+Hafiidz@users.noreply.github.com>
Date: Fri, 9 Sep 2022 19:07:35 +0800
Subject: [PATCH] Urgent Fix to PR:860

---
 webui-streamlit.cmd | 3 ++-
 webui.cmd           | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/webui-streamlit.cmd b/webui-streamlit.cmd
index 6ba0597..832689b 100644
--- a/webui-streamlit.cmd
+++ b/webui-streamlit.cmd
@@ -2,7 +2,8 @@
 
 :: copy over the first line from environment.yaml, e.g. name: ldm, and take the second word after splitting by ":" delimiter 
 set /p first_line=< environment.yaml  
-for /f "tokens=2 delims=:" %%i in ("%first_line%") do set conda_env_name=%%i
+for /f "tokens=2 delims=:" %%i in ("%first_line%") do set untrimmed_conda_env_name=%%i
+for /f "tokens=* delims= " %%a in ("%untrimmed_conda_env_name%") do set conda_env_name=%%a
 echo Environment name is set as %conda_env_name% as per environment.yaml
 
 :: Put the path to conda directory after "=" sign if it's installed at non-standard path:
diff --git a/webui.cmd b/webui.cmd
index 9ba6695..ac4d61f 100644
--- a/webui.cmd
+++ b/webui.cmd
@@ -2,7 +2,8 @@
 
 :: copy over the first line from environment.yaml, e.g. name: ldm, and take the second word after splitting by ":" delimiter 
 set /p first_line=< environment.yaml  
-for /f "tokens=2 delims=:" %%i in ("%first_line%") do set conda_env_name=%%i
+for /f "tokens=2 delims=:" %%i in ("%first_line%") do set untrimmed_conda_env_name=%%i
+for /f "tokens=* delims= " %%a in ("%untrimmed_conda_env_name%") do set conda_env_name=%%a
 echo Environment name is set as %conda_env_name% as per environment.yaml
 
 :: Put the path to conda directory after "=" sign if it's installed at non-standard path: