update to the latest plugin format

tuneflow · Feb 28, 2023 · 0048080 · 0048080
1 parent 2316ef7
commit 0048080
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/plugin.py b/plugin.py
@@ -1,9 +1,8 @@
-from tuneflow_py import TuneflowPlugin, ParamDescriptor, Song, ReadAPIs, TrackType, WidgetType, LabelText
+from tuneflow_py import TuneflowPlugin, ParamDescriptor, Song, TrackType, WidgetType, LabelText
 from typing import Dict, Any
 from audioldm import text_to_audio, build_model
 from pathlib import Path
 import traceback
-import random
 from io import BytesIO
 import soundfile as sf
 from typing import List
@@ -26,7 +25,7 @@ def provider_display_name() -> LabelText:
         }
 
     @staticmethod
-    def params(song: Song, read_apis: ReadAPIs) -> Dict[str, ParamDescriptor]:
+    def params(song: Song) -> Dict[str, ParamDescriptor]:
         return {
             "prompt": {
                 "displayName": {
@@ -105,7 +104,7 @@ def params(song: Song, read_apis: ReadAPIs) -> Dict[str, ParamDescriptor]:
         }
 
     @staticmethod
-    def run(song: Song, params: Dict[str, Any], read_apis: ReadAPIs):
+    def run(song: Song, params: Dict[str, Any]):
         model_path = str(Path(__file__).parent.joinpath('ckpt/ldm_trimmed.ckpt').absolute())
         model = build_model(ckpt_path=model_path)
         # TODO: Support prompt i18n

diff --git a/requirements.txt b/requirements.txt
@@ -13,4 +13,4 @@ pandas
 torchlibrosa==0.0.9
 transformers
 ftfy
-tuneflow-py>=0.1.2
+tuneflow-py>=0.3.0