Update app.py
Browse files
app.py
CHANGED
|
@@ -137,12 +137,6 @@ def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=
|
|
| 137 |
|
| 138 |
# Set up environment variables for CUDA with optimized settings
|
| 139 |
env = os.environ.copy()
|
| 140 |
-
env.update({
|
| 141 |
-
"CUDA_VISIBLE_DEVICES": "0",
|
| 142 |
-
"CUDA_HOME": "/usr/local/cuda",
|
| 143 |
-
"PATH": f"/usr/local/cuda/bin:{env.get('PATH', '')}",
|
| 144 |
-
"LD_LIBRARY_PATH": f"/usr/local/cuda/lib64:{env.get('LD_LIBRARY_PATH', '')}"
|
| 145 |
-
})
|
| 146 |
|
| 147 |
# Execute the command
|
| 148 |
try:
|
|
|
|
| 137 |
|
| 138 |
# Set up environment variables for CUDA with optimized settings
|
| 139 |
env = os.environ.copy()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 140 |
|
| 141 |
# Execute the command
|
| 142 |
try:
|