phuochungus commited on
Commit
56c1354
1 Parent(s): bee6ac4

Add model download script and update compose.yaml

Browse files
Files changed (2) hide show
  1. compose.yaml +1 -0
  2. model/download.sh +21 -0
compose.yaml CHANGED
@@ -14,3 +14,4 @@ services:
14
  - NEO4J_PASSWORD=your_neo4j_password
15
  - AURA_INSTANCEID=your_neo4j_aura_instance_id
16
  - AURA_INSTANCENAME=your_neo4j_aura_instance_name
 
 
14
  - NEO4J_PASSWORD=your_neo4j_password
15
  - AURA_INSTANCEID=your_neo4j_aura_instance_id
16
  - AURA_INSTANCENAME=your_neo4j_aura_instance_name
17
+ - MODEL_URL=https://hdfxssmjuydwfwarxnfe.supabase.co/storage/v1/object/public/model/best20231112.onnx
model/download.sh ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ cd "$(dirname "$0")"
4
+
5
+ EXPECTED_CHECKSUM="1282ca01d9dd18fa3632eb6ea27fbfe6"
6
+ ACTUAL_CHECKSUM=$(md5sum best20231112.onnx | awk '{print $1}')
7
+
8
+ if [ "$ACTUAL_CHECKSUM" != "$EXPECTED_CHECKSUM" ]; then
9
+ echo "Checksum mismatch, downloading model"
10
+ if [ -n "$MODEL_URL" ]; then
11
+ echo "Downloading model"
12
+ wget -O best20231112.onnx $MODEL_URL
13
+ exit 0
14
+ else
15
+ echo "ERROR: MODEL_URL not set"
16
+ exit 1
17
+ fi
18
+ else
19
+ echo "Model already downloaded"
20
+ exit 0
21
+ fi