kadirnar commited on
Commit
7f75ea6
·
verified ·
1 Parent(s): 8eda1d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import gradio as gr
2
  import os
3
  from vyvotts.audio_tokenizer import process_dataset
 
4
 
 
5
  def process_dataset_ui(
6
  original_dataset,
7
  output_dataset,
@@ -10,7 +12,7 @@ def process_dataset_ui(
10
  hf_token
11
  ):
12
  """
13
- Process dataset with Gradio UI.
14
 
15
  Args:
16
  original_dataset: HuggingFace dataset path to process
 
1
  import gradio as gr
2
  import os
3
  from vyvotts.audio_tokenizer import process_dataset
4
+ import spaces
5
 
6
+ @spaces.GPU(duration=120)
7
  def process_dataset_ui(
8
  original_dataset,
9
  output_dataset,
 
12
  hf_token
13
  ):
14
  """
15
+ Process dataset with Gradio UI using ZeroGPU.
16
 
17
  Args:
18
  original_dataset: HuggingFace dataset path to process