justus-tobias commited on
Commit
3d2b6af
1 Parent(s): 623f68f

utilising ZeroGPU

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +2 -1
app.py CHANGED
@@ -3,10 +3,10 @@ import torch
3
  from huggingface_hub import hf_hub_download
4
  from moshi.models import loaders, LMGen
5
  import numpy as np
 
6
 
7
 
8
-
9
-
10
  def process_wav_new(in_wav):
11
  """wav = torch.randn(1, 1, 24000 * 10) # should be [B, C=1, T]"""
12
 
@@ -156,4 +156,4 @@ Monologue” method significantly improves the linguistic quality of generated s
156
  elem_id="citation-button",
157
  show_copy_button=True,
158
  )
159
- demo.launch(debug=True, share=True)
 
3
  from huggingface_hub import hf_hub_download
4
  from moshi.models import loaders, LMGen
5
  import numpy as np
6
+ import spaces
7
 
8
 
9
+ @spaces.GPU
 
10
  def process_wav_new(in_wav):
11
  """wav = torch.randn(1, 1, 24000 * 10) # should be [B, C=1, T]"""
12
 
 
156
  elem_id="citation-button",
157
  show_copy_button=True,
158
  )
159
+ demo.launch(debug=True)
requirements.txt CHANGED
@@ -3,4 +3,5 @@ gradio==4.44.0
3
  triton==3.0.0
4
  torchaudio>=2.2.0+cu121
5
  --extra-index-url https://download.pytorch.org/whl/cu113
6
- torch
 
 
3
  triton==3.0.0
4
  torchaudio>=2.2.0+cu121
5
  --extra-index-url https://download.pytorch.org/whl/cu113
6
+ torch
7
+ spaces