Spaces:
Running
Running
jhj0517
commited on
Commit
·
afc1f96
1
Parent(s):
813ca79
Add yaml utils
Browse files
modules/whisper/whisper_parameter.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from dataclasses import dataclass, fields
|
2 |
import gradio as gr
|
3 |
-
from typing import Optional
|
4 |
import yaml
|
5 |
|
6 |
|
@@ -277,7 +277,7 @@ class WhisperValues:
|
|
277 |
A data class to use Whisper parameters.
|
278 |
"""
|
279 |
|
280 |
-
def to_yaml(self) ->
|
281 |
data = {
|
282 |
"whisper": {
|
283 |
"model_size": self.model_size,
|
@@ -325,4 +325,4 @@ class WhisperValues:
|
|
325 |
"hf_token": self.hf_token
|
326 |
}
|
327 |
}
|
328 |
-
return
|
|
|
1 |
from dataclasses import dataclass, fields
|
2 |
import gradio as gr
|
3 |
+
from typing import Optional, Dict
|
4 |
import yaml
|
5 |
|
6 |
|
|
|
277 |
A data class to use Whisper parameters.
|
278 |
"""
|
279 |
|
280 |
+
def to_yaml(self) -> Dict:
|
281 |
data = {
|
282 |
"whisper": {
|
283 |
"model_size": self.model_size,
|
|
|
325 |
"hf_token": self.hf_token
|
326 |
}
|
327 |
}
|
328 |
+
return data
|