Modified the bash script

tapis-project · Aug 2, 2024 · 8bb9701 · 8bb9701
1 parent 74a28ef
commit 8bb9701
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/external_plugins/image_scoring_plugin/entrypoint-nano.sh b/external_plugins/image_scoring_plugin/entrypoint-nano.sh
@@ -1,16 +1,16 @@
 #!/bin/bash
 echo "Start of Image Scoring Plugin Nano..."
-if [ "$MODEL_TYPE" -eq 1 ]; then
+if [ $MODEL_TYPE -eq 1 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/md_v5b.0.0.pt
   echo "Executing the MDV5B second model. Downloading the model..."
 fi
 
-if [ "$MODEL_TYPE" -eq 2 ]; then
+if [ $MODEL_TYPE -eq 2 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/md_v5a.0.0_ena.pt
   echo "Executing the Megadetector fine-tuned ENA model. Downloading the model..."
 fi
 
-if [ "$MODEL_TYPE" -eq 3 ]; then
+if [ $MODEL_TYPE -eq 3 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/mdv5_optimized.pt
   echo "Executing the Megadetector optimized model. Downloading the model..."
 fi

diff --git a/external_plugins/image_scoring_plugin/entrypoint.sh b/external_plugins/image_scoring_plugin/entrypoint.sh
@@ -1,16 +1,16 @@
 #!/bin/bash
 echo "Start of Image Scoring Plugin..."
-if [ "$MODEL_TYPE" -eq 1 ]; then
+if [ $MODEL_TYPE -eq 1 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/md_v5b.0.0.pt
   echo "Executing the MDV5B second model. Downloading the model..."
 fi
 
-if [ "$MODEL_TYPE" -eq 2 ]; then
+if [ $MODEL_TYPE -eq 2 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/md_v5a.0.0_ena.pt
   echo "Executing the Megadetector fine-tuned ENA model. Downloading the model..."
 fi
 
-if [ "$MODEL_TYPE" -eq 3 ]; then
+if [ $MODEL_TYPE -eq 3 ]; then
   wget -q -O /md_v5a.0.0.pt https://github.com/ICICLE-ai/camera_traps/raw/main/models/mdv5_optimized.pt
   echo "Executing the Megadetector optimized model. Downloading the model..."
 fi