Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -19,7 +19,7 @@ ALLOWED_EXTENSIONS = {'wav', 'mp3', 'mp4', 'mpeg', 'mpga', 'm4a', 'webm', 'flac'
|
|
19 |
|
20 |
# Initialize model variable
|
21 |
model = None
|
22 |
-
MODEL_SIZE = "
|
23 |
|
24 |
# Set cache directory for Hugging Face Spaces
|
25 |
def setup_cache_directory():
|
|
|
19 |
|
20 |
# Initialize model variable
|
21 |
model = None
|
22 |
+
MODEL_SIZE = "medium" # Use 'tiny' for faster loading on Hugging Face Spaces, change to 'base', 'small', 'medium', or 'large' as needed
|
23 |
|
24 |
# Set cache directory for Hugging Face Spaces
|
25 |
def setup_cache_directory():
|