Spaces:
Runtime error
Runtime error
phuochungus
commited on
Commit
•
56c1354
1
Parent(s):
bee6ac4
Add model download script and update compose.yaml
Browse files- compose.yaml +1 -0
- 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
|