NariLabs commited on
Commit
2a66429
·
verified ·
1 Parent(s): b668a3e

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -8,6 +8,7 @@ from typing import List, Tuple
8
 
9
  import gradio as gr
10
  import torch
 
11
 
12
  from dia2 import Dia2, GenerationConfig, SamplingConfig
13
 
@@ -99,6 +100,7 @@ def _prepare_prefix(file_path: str | None) -> str | None:
99
  return str(path)
100
 
101
 
 
102
  def generate_audio(
103
  turn_count: int,
104
  *inputs,
 
8
 
9
  import gradio as gr
10
  import torch
11
+ import spaces
12
 
13
  from dia2 import Dia2, GenerationConfig, SamplingConfig
14
 
 
100
  return str(path)
101
 
102
 
103
+ @spaces.GPU(duration=100)
104
  def generate_audio(
105
  turn_count: int,
106
  *inputs,