fix: move to new definition

BIGmama-technology · Mar 9, 2024 · c10f48f · c10f48f
1 parent 86d0a19
commit c10f48f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/hyko_toolkit/models/audio/automatic_speech_recognition/metadata.py b/hyko_toolkit/models/audio/automatic_speech_recognition/metadata.py
@@ -1,10 +1,12 @@
 from pydantic import Field
 
-from hyko_sdk.function import SDKFunction
+from hyko_sdk.definitions import ToolkitModel
 from hyko_sdk.io import Audio
 from hyko_sdk.metadata import CoreModel
 
-func = SDKFunction(
+func = ToolkitModel(
+    name="automatic_speech_recognition",
+    task="audio",
     description="HuggingFace automatic speech recognition",
 )
 

diff --git a/hyko_toolkit/models/audio/text_to_speech/metadata.py b/hyko_toolkit/models/audio/text_to_speech/metadata.py
@@ -1,11 +1,13 @@
 from pydantic import Field
 
-from hyko_sdk.function import SDKFunction
+from hyko_sdk.definitions import ToolkitModel
 from hyko_sdk.io import Audio
 from hyko_sdk.metadata import CoreModel
 
-func = SDKFunction(
-    description="HuggingFace text to speech, run on cude may cause issues on cpu",
+func = ToolkitModel(
+    name="text_to_speech",
+    task="audio",
+    description="HuggingFace text to speech, run on cuda may cause issues on cpu",
 )