Spaces:
Runtime error
Runtime error
Commit
·
f28703e
1
Parent(s):
447a9ff
fixcpu
Browse files- src/webui.py +1 -1
src/webui.py
CHANGED
@@ -148,7 +148,7 @@ def process_file_upload(file):
|
|
148 |
|
149 |
|
150 |
if __name__ == '__main__':
|
151 |
-
os.
|
152 |
parser = ArgumentParser(description='Generate a AI cover song in the song_output/id directory.', add_help=True)
|
153 |
parser.add_argument("--share", action="store_true", dest="share_enabled", default=False, help="Enable sharing")
|
154 |
parser.add_argument("--listen", action="store_true", default=False, help="Make the WebUI reachable from your local network.")
|
|
|
148 |
|
149 |
|
150 |
if __name__ == '__main__':
|
151 |
+
os.system("pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu")
|
152 |
parser = ArgumentParser(description='Generate a AI cover song in the song_output/id directory.', add_help=True)
|
153 |
parser.add_argument("--share", action="store_true", dest="share_enabled", default=False, help="Enable sharing")
|
154 |
parser.add_argument("--listen", action="store_true", default=False, help="Make the WebUI reachable from your local network.")
|