From 0a74e8176591681d4cd62070bca7f6c4fa0d4fd2 Mon Sep 17 00:00:00 2001
From: Francesco Caracciolo
 <67018178+FrancescoCaracciolo@users.noreply.github.com>
Date: Sat, 7 Dec 2024 16:38:57 +0100
Subject: [PATCH] Add extra instructions for ollama

---
 src/dataset.py | 12 +++++++++++-
 src/tts.py     |  3 ++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/dataset.py b/src/dataset.py
index 3be23fe..4e0ef9e 100644
--- a/src/dataset.py
+++ b/src/dataset.py
@@ -197,6 +197,11 @@ def reconstruct_dataset_from_csv(filename):
 ```console
 sudo pacman -S ollama
 ```
+If you have a Nvidia GPU, remember to install ollama-cuda for better performances:
+```console 
+sudo pacman -S ollama-cuda
+```
+
 Then you can run ollama by executing this command:
 ```console
 ollama serve
@@ -205,7 +210,12 @@ def reconstruct_dataset_from_csv(filename):
 ```console
 ollama pull llama3.1
 ```
-("""
+
+You can also install custom models from HuggingFace:
+```console
+ollama pull hf.co/{username}/{repository}
+```
+"""
 }
 
 if is_flatpak():
diff --git a/src/tts.py b/src/tts.py
index e518e3a..fdb16ae 100644
--- a/src/tts.py
+++ b/src/tts.py
@@ -110,7 +110,8 @@ def get_current_voice(self):
         voice = self.get_setting("voice")
         if not voice:
             if not self.voices:
-                return None return self.voices[0][1]
+                return None
+            return self.voices[0][1]
         else:
             return voice