Spaces:
Runtime error
Runtime error
Commit ·
65066a0
1
Parent(s): 22826d0
Fix NumPy compatibility issue and improve filename handling
Browse files- requirements.txt +2 -1
- server.py +12 -3
requirements.txt
CHANGED
|
@@ -5,4 +5,5 @@ Werkzeug==2.3.6
|
|
| 5 |
gunicorn==21.2.0
|
| 6 |
soundfile==0.12.1
|
| 7 |
torch==2.0.1
|
| 8 |
-
torchaudio==2.0.2
|
|
|
|
|
|
| 5 |
gunicorn==21.2.0
|
| 6 |
soundfile==0.12.1
|
| 7 |
torch==2.0.1
|
| 8 |
+
torchaudio==2.0.2
|
| 9 |
+
numpy<2.0
|
server.py
CHANGED
|
@@ -66,8 +66,17 @@ def separate_audio():
|
|
| 66 |
temp_dir = tempfile.mkdtemp(prefix=f"demucs_{session_id}_")
|
| 67 |
|
| 68 |
try:
|
| 69 |
-
# Save uploaded file to temp directory
|
| 70 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
file_path = os.path.join(temp_dir, filename)
|
| 72 |
file.save(file_path)
|
| 73 |
|
|
@@ -138,7 +147,7 @@ def separate_audio():
|
|
| 138 |
# Store session data in memory
|
| 139 |
active_sessions[session_id] = {
|
| 140 |
'tracks': track_data,
|
| 141 |
-
'original_filename':
|
| 142 |
'created_at': time.time(),
|
| 143 |
'temp_dir': temp_dir
|
| 144 |
}
|
|
|
|
| 66 |
temp_dir = tempfile.mkdtemp(prefix=f"demucs_{session_id}_")
|
| 67 |
|
| 68 |
try:
|
| 69 |
+
# Save uploaded file to temp directory with clean filename
|
| 70 |
+
original_filename = secure_filename(file.filename)
|
| 71 |
+
# Remove any problematic characters and ensure proper extension
|
| 72 |
+
base_name = os.path.splitext(original_filename)[0]
|
| 73 |
+
extension = os.path.splitext(original_filename)[1].lower()
|
| 74 |
+
|
| 75 |
+
# Clean the base name to avoid issues
|
| 76 |
+
import re
|
| 77 |
+
clean_base = re.sub(r'[^\w\-_\.]', '_', base_name)
|
| 78 |
+
filename = f"{clean_base}{extension}"
|
| 79 |
+
|
| 80 |
file_path = os.path.join(temp_dir, filename)
|
| 81 |
file.save(file_path)
|
| 82 |
|
|
|
|
| 147 |
# Store session data in memory
|
| 148 |
active_sessions[session_id] = {
|
| 149 |
'tracks': track_data,
|
| 150 |
+
'original_filename': original_filename, # Use original filename for display
|
| 151 |
'created_at': time.time(),
|
| 152 |
'temp_dir': temp_dir
|
| 153 |
}
|