klentyboopathi commited on
Commit
0186ea0
·
1 Parent(s): 554c407
Files changed (2) hide show
  1. bot/bot_websocket_server.py +2 -2
  2. setup.sh +5 -5
bot/bot_websocket_server.py CHANGED
@@ -89,14 +89,14 @@ async def run_bot_websocket_server(websocket_client):
89
  )
90
 
91
  stt = WhisperSTTService(
92
- model="tiny",
93
  device="cpu",
94
  compute_type="default",
95
  language="en",
96
  )
97
 
98
  llm = OLLamaLLMService(
99
- model="llama3.1",
100
  # params=OLLamaLLMService.InputParams(temperature=0.7, max_tokens=1000),
101
  )
102
 
 
89
  )
90
 
91
  stt = WhisperSTTService(
92
+ model="smollm",
93
  device="cpu",
94
  compute_type="default",
95
  language="en",
96
  )
97
 
98
  llm = OLLamaLLMService(
99
+ model="smoll",
100
  # params=OLLamaLLMService.InputParams(temperature=0.7, max_tokens=1000),
101
  )
102
 
setup.sh CHANGED
@@ -1,9 +1,9 @@
1
- mkdir -p bot/assets
2
- wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/kokoro-v1.0.onnx -O bot/assets/kokoro-v1.0.int8.onnx
3
- wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/kokoro-v1.0.fp16-gpu.onnx -O bot/assets/kokoro-v1.0.fp16-gpu.onnx
4
- wget https://huggingface.co/NeuML/kokoro-base-onnx/resolve/main/voices.json -O bot/assets/voices.json
5
  curl -fsSL https://ollama.com/install.sh | sh
6
  sleep 100
7
  ollama serve &
8
  sleep 100
9
- ollama pull llama3.1
 
1
+ # mkdir -p bot/assets
2
+ # wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/kokoro-v1.0.onnx -O bot/assets/kokoro-v1.0.int8.onnx
3
+ # wget https://github.com/thewh1teagle/kokoro-onnx/releases/download/model-files-v1.0/kokoro-v1.0.fp16-gpu.onnx -O bot/assets/kokoro-v1.0.fp16-gpu.onnx
4
+ # wget https://huggingface.co/NeuML/kokoro-base-onnx/resolve/main/voices.json -O bot/assets/voices.json
5
  curl -fsSL https://ollama.com/install.sh | sh
6
  sleep 100
7
  ollama serve &
8
  sleep 100
9
+ ollama pull smollm