Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import sys
|
| 2 |
import os
|
| 3 |
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'amt/src')))
|
|
@@ -10,7 +13,6 @@ from html_helper import *
|
|
| 10 |
from model_helper import *
|
| 11 |
|
| 12 |
from pytube import YouTube
|
| 13 |
-
import torch
|
| 14 |
import torchaudio
|
| 15 |
import glob
|
| 16 |
import gradio as gr
|
|
@@ -19,7 +21,7 @@ import gradio as gr
|
|
| 19 |
|
| 20 |
# @title Load Checkpoint
|
| 21 |
model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
| 22 |
-
precision = '16' if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
|
| 23 |
project = '2024'
|
| 24 |
|
| 25 |
if model_name == "YMT3+":
|
|
|
|
| 1 |
+
|
| 2 |
+
import spaces
|
| 3 |
+
|
| 4 |
import sys
|
| 5 |
import os
|
| 6 |
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'amt/src')))
|
|
|
|
| 13 |
from model_helper import *
|
| 14 |
|
| 15 |
from pytube import YouTube
|
|
|
|
| 16 |
import torchaudio
|
| 17 |
import glob
|
| 18 |
import gradio as gr
|
|
|
|
| 21 |
|
| 22 |
# @title Load Checkpoint
|
| 23 |
model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
|
| 24 |
+
precision = '16'# if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
|
| 25 |
project = '2024'
|
| 26 |
|
| 27 |
if model_name == "YMT3+":
|