Upload 2 files

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import datetime
 import pytz
+from pathlib import Path
 
 def current_time():
 	current = datetime.datetime.now(pytz.timezone('Asia/Shanghai')).strftime("%Y年-%m月-%d日 %H时:%M分:%S秒")
@@ -9,7 +10,6 @@ def current_time():
 
 print(f"[{current_time()}] 开始部署空间...")
 
-from pathlib import Path
 print(f"[{current_time()}] 日志：安装 - 必要包")
 os.system("pip install -r ./requirements.txt")
 print(f"[{current_time()}] 日志：安装 - gsutil")
@@ -25,6 +25,14 @@ os.system("python3 -m pip install -e .")
 print(f"[{current_time()}] 日志：Python - 更新 Python 包管理器 pip")
 os.system("python3 -m pip install --upgrade pip")
 
+# 安装 airio
+print(f"[{current_time()}] 日志：Git - 克隆 Github 的 airio 到当前目录")
+os.system("git clone --branch=main https://github.com/google/airio")
+print(f"[{current_time()}] 日志：文件 - 移动 airio 到当前目录并重命名为 airio_tmp 并删除")
+os.system("mv airio airio_tmp; mv airio_tmp/* .; rm -r airio_tmp")
+print(f"[{current_time()}] 日志：Python - 使用 pip 安装 当前目录内的 Python 包")
+os.system("python3 -m pip install -e .")
+
 # 安装 mt3
 print(f"[{current_time()}] 日志：Git - 克隆 Github 的 MT3 模型到当前目录")
 os.system("git clone --branch=main https://github.com/magenta/mt3")
@@ -43,14 +51,6 @@ os.system("gsutil -q -m cp -r gs://mt3/checkpoints .")
 print(f"[{current_time()}] 日志：gsutil - 复制 SoundFont 文件到当前目录")
 os.system("gsutil -q -m cp gs://magentadata/soundfonts/SGM-v2.01-Sal-Guit-Bass-V1.3.sf2 .")
 
-# 安装 airio
-print(f"[{current_time()}] 日志：Git - 克隆 Github 的 airio 到当前目录")
-os.system("git clone --branch=main https://github.com/google/airio")
-print(f"[{current_time()}] 日志：文件 - 移动 airio 到当前目录并重命名为 airio_tmp 并删除")
-os.system("mv airio airio_tmp; mv airio_tmp/* .; rm -r airio_tmp")
-print(f"[{current_time()}] 日志：Python - 使用 pip 安装 当前目录内的 Python 包")
-os.system("python3 -m pip install -e .")
-
 #@title 导入和定义
 print(f"[{current_time()}] 日志：导入 - 必要工具")
 import functools
@@ -110,7 +110,7 @@ class InferenceModel(object):
 		self.batch_size = 8
 		self.outputs_length = 1024
 		self.sequence_length = {'inputs': self.inputs_length,
-														'targets': self.outputs_length}
+								'targets': self.outputs_length}
 
 		self.partitioner = t5x.partitioning.PjitPartitioner(
 				model_parallel_submesh=None, num_partitions=1)
@@ -120,7 +120,8 @@ class InferenceModel(object):
 		self.spectrogram_config = spectrograms.SpectrogramConfig()
 		self.codec = vocabularies.build_codec(
 				vocab_config=vocabularies.VocabularyConfig(
-						num_velocity_bins=num_velocity_bins))
+				num_velocity_bins=num_velocity_bins)
+				)
 		self.vocabulary = vocabularies.vocabulary_from_codec(self.codec)
 		self.output_features = {
 				'inputs': seqio.ContinuousFeature(dtype=tf.float32, rank=2),
@@ -314,14 +315,14 @@ article = "<p style='text-align: center'>出错了？试试把文件转换为MP3
 examples=[['canon.flac'], ['download.wav']]
 
 gr.Interface(
-		inference,
-		gr.inputs.Audio(type="filepath", label="输入"),
-		[gr.outputs.File(label="输出")],
-		title=title,
-		description=description,
-		article=article,
-		examples=examples,
-		allow_flagging=False,
-		allow_screenshot=False,
-		enable_queue=True
-		).launch()
+	inference,
+	gr.inputs.Audio(type="filepath", label="输入"),
+	[gr.outputs.File(label="输出")],
+	title=title,
+	description=description,
+	article=article,
+	examples=examples,
+	allow_flagging=False,
+	allow_screenshot=False,
+	enable_queue=True
+	).launch()

requirements.txt

@@ -1,19 +1,19 @@
 nest-asyncio
 pyfluidsynth
-absl-py 
-ddsp 
-flax 
-gin-config 
-immutabledict 
-librosa 
-mir_eval 
-note_seq 
-numpy 
-pretty_midi 
-scikit-learn 
-scipy 
-seqio 
-t5 
+absl-py
+ddsp
+flax
+gin-config
+gradio
+immutabledict
+librosa
+mir_eval
+note_seq
+numpy
+pretty_midi
+scikit-learn
+scipy
+seqio
 tensorflow_cpu
-tensorflow-datasets 
-
+tensorflow-datasets
+t5