Fixes and script upgrades

Signed-off-by: Rishi Banerjee <[email protected]>

main.py

@@ -23,6 +23,8 @@
 from mem_test import measure_memory_usage
 
 nltk.download('averaged_perceptron_tagger')
+nltk.download('punkt')
+
 
 
 def timing(f):

requirements.txt

@@ -5,4 +5,5 @@ pydub
 python-dotenv
 fuzzywuzzy
 python-Levenshtein
-nltk
+nltk
+psutil

setup.sh

@@ -1,15 +1,15 @@
 #!/bin/bash
-sudo sed -i '1i\deb http://mirror.mythic-beasts.com/raspbian/raspbian buster main contrib non-free rpi' /etc/apt/sources.list
+# sudo sed -i '1i\deb http://mirror.mythic-beasts.com/raspbian/raspbian buster main contrib non-free rpi' /etc/apt/sources.list
 
 # Comment out other lines if not already commented
-sudo sed -i -e '2,$s/^deb/#deb/' /etc/apt/sources.list
+# sudo sed -i -e '2,$s/^deb/#deb/' /etc/apt/sources.list
 
 # Update package lists and upgrade installed packages
-sudo apt-get update
-sudo apt-get upgrade -y
+# sudo apt-get update
+# sudo apt-get upgrade -y
 sudo apt install git
 # Install required packages
-sudo apt-get install -y python3-pyaudio flac libsdl-ttf2.0-0 python3-sdl2 libsdl2-mixer-2.0-0
+sudo apt-get install -y python3-pyaudio flac libsdl-ttf2.0-0 python3-sdl2 libsdl2-mixer-2.0-0 ffmpeg
 sudo apt-get install python3-pygame
 
 # Install Python Installer