Upload app.py

app.py

@@ -7,11 +7,23 @@ import datetime
 import subprocess
 import os
 from inference import postprocess_inst_names
-
-# 如果你的 inference、convert 等逻辑和原来一致，可以直接用
 from inference import inference_patch
 from convert import abc2xml, xml2, pdf2img
 
+try:
+    import spaces
+    USING_SPACES = True
+except ImportError:
+    USING_SPACES = False
+import zero
+
+
+def gpu_decorator(func):
+    if USING_SPACES:
+        return spaces.GPU(func)
+    else:
+        return func
+
 
 # 读取 prompt 组合
 with open('prompts.txt', 'r') as f:
@@ -141,6 +153,7 @@ def update_page(direction, data):
     return new_image, prev_btn_state, next_btn_state, data
 
 
+@gpu_decorator
 def generate_music(period, composer, instrumentation):
     """
     需要保证每次 yield 的返回值数量一致。