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) -> str:
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 yaml.dump(data, sort_keys=False, default_flow_style=False)
 
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