Update app.py

app.py

@@ -1,3 +1,6 @@
+
+import spaces
+
 import sys
 import os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'amt/src')))
@@ -10,7 +13,6 @@ from html_helper import *
 from model_helper import *
 
 from pytube import YouTube
-import torch
 import torchaudio
 import glob
 import gradio as gr
@@ -19,7 +21,7 @@ import gradio as gr
 
 # @title Load Checkpoint
 model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
-precision = '16' if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
+precision = '16'# if torch.cuda.is_available() else '32'# @param ["32", "bf16-mixed", "16"]
 project = '2024'
 
 if model_name == "YMT3+":