Update readme

README.md

@@ -44,14 +44,20 @@ If you only want to download the required files to use the model (without clonin
 Here's an example:
 
 ```python
+import os
+import sys
+
 from huggingface_hub import hf_hub_download
 import importlib.util
 
 # Specify the repository
 repo_id = "tlmk22/QuefrencyGuardian"
 
-# Download the Python file containing the model class
+# Download the Python file containing the model class and add it to your path
 model_path = hf_hub_download(repo_id=repo_id, filename="model.py")
+model_dir = os.path.dirname(model_path)  
+if model_dir not in sys.path:
+    sys.path.append(model_dir)
 
 # Dynamically load the class from the downloaded file
 spec = importlib.util.spec_from_file_location("model", model_path)
@@ -65,7 +71,7 @@ FastModelHuggingFace = model_module.FastModelHuggingFace
 fast_model = FastModelHuggingFace.from_pretrained(repo_id)
 
 # Perform predictions
-result = fast_model.predict("path/to/audio.wav")
+result = fast_model.predict("path/to/audio.wav", device="cpu")
 map_labels = {0: "chainsaw", 1: "environment"}
 print(f"Prediction Result: {map_labels[result[0]]}")
 ```