From dcba1b2800af3fc62f9c4d7e3a4a3bcd39846034 Mon Sep 17 00:00:00 2001 From: Wiwi Kuan Date: Fri, 4 Oct 2024 09:51:43 +0800 Subject: [PATCH] Add files via upload --- nonsense.sh | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 nonsense.sh diff --git a/nonsense.sh b/nonsense.sh new file mode 100644 index 0000000..8190c42 --- /dev/null +++ b/nonsense.sh @@ -0,0 +1,44 @@ +#!/bin/bash + +generate_story() { + local model=$1 + local prompt="Write a long nonsensical story." + ollama run "$model" "$prompt" +} + +save_story() { + local story=$1 + local index=$2 + local filename="nonsense_story_$index.txt" + echo "$story" > "$filename" + echo "Saved story to $filename" +} + +print_usage() { + echo "Usage: $0 " + echo "Example: $0 5 llama2" +} + +if [ $# -ne 2 ]; then + print_usage + exit 1 +fi + +num_stories=$1 +model=$2 + +if ! [[ "$num_stories" =~ ^[0-9]+$ ]]; then + echo "Error: Number of stories must be a positive integer." + print_usage + exit 1 +fi + +echo "Generating $num_stories nonsense stories using the $model model..." + +for i in $(seq 1 $num_stories); do + echo -e "\nGenerating story $i..." + story=$(generate_story "$model") + save_story "$story" "$i" +done + +echo -e "\nAll stories generated and saved!"