wangmengchao commited on
Commit
75412c3
·
1 Parent(s): d5c68c9
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from pathlib import Path
3
- import argparse
4
  from datetime import datetime
5
  import librosa
6
  from infer import load_models,main
@@ -12,6 +12,7 @@ try:
12
  _ = torch.tensor([0.0]).to('cuda')
13
  except Exception as e:
14
  print(f"GPU warmup failed: {e}")
 
15
 
16
  pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
17
  @spaces.GPU
 
1
  import gradio as gr
2
  from pathlib import Path
3
+ import argparse,os
4
  from datetime import datetime
5
  import librosa
6
  from infer import load_models,main
 
12
  _ = torch.tensor([0.0]).to('cuda')
13
  except Exception as e:
14
  print(f"GPU warmup failed: {e}")
15
+ os.environ["GRADIO_TEMP_DIR"] = "./tmp"
16
 
17
  pipe,fantasytalking,wav2vec_processor,wav2vec = None,None,None,None
18
  @spaces.GPU