Update install.sh

install.sh

@@ -8,49 +8,38 @@ if ! command -v ollama &> /dev/null; then
     exit 1
 fi
 
-# Download model files
-MODEL_DIR="$HOME/.ollama/models/indo_math_teacher"
-mkdir -p "$MODEL_DIR"
+# Create directory for model files
+mkdir -p model_files
 
+# Download model files
 echo "Downloading model files..."
-FILES=(
-    "added_tokens.json"
-    "config.json"
-    "merges.txt"
-    "model.safetensors"
-    "model.safetensors.index.json"
-    "special_tokens_map.json"
-    "tokenizer.json"
-    "tokenizer_config.json"
-    "vocab.json"
-)
-
-HF_REPO="Lckoo1230/indo-math-teacher-complete"
-
-for file in "${FILES[@]}"; do
-    echo "Downloading $file..."
-    curl -L "https://huggingface.co/$HF_REPO/resolve/main/$file" -o "$MODEL_DIR/$file"
-done
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/added_tokens.json -o model_files/added_tokens.json
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/config.json -o model_files/config.json
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/merges.txt -o model_files/merges.txt
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/model.safetensors -o model_files/model.safetensors
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/special_tokens_map.json -o model_files/special_tokens_map.json
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/tokenizer.json -o model_files/tokenizer.json
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/tokenizer_config.json -o model_files/tokenizer_config.json
+curl -L https://huggingface.co/Lckoo1230/indo-math-teacher-complete/resolve/main/vocab.json -o model_files/vocab.json
 
 # Create Modelfile
-cat > Modelfile.temp << EOL
-FROM qwen/Qwen2.5-1.5B
+cat > Modelfile_Indo_Math_Teacher << EOL
+FROM Qwen/Qwen2.5-Math-1.5B
 SYSTEM "You are a math teacher using the Gasing method"
 
 PARAMETER temperature 0.1
 PARAMETER top_p 0.9
 PARAMETER num_predict 200
 
-# Set stop sequences
-TEMPLATE "{{ .System }}\n\n{{ .Prompt }}"
+TEMPLATE "{{.System}}\n\n{{.Prompt}}"
 STOP "[/INST]"
 STOP ">>> "
 STOP "\n\nHuman:"
 STOP "\nHuman:"
 EOL
 
-# Create the model in Ollama
-ollama create indo_math_teacher -f Modelfile.temp
+# Create the Ollama model
+ollama create indo_math_teacher -f Modelfile_Indo_Math_Teacher
 
 echo "Indo Math Teacher installed successfully!"
 echo "Run with: ollama run indo_math_teacher"