junxiliu commited on
Commit
b919870
·
1 Parent(s): 6e47a03
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ if torch.cuda.is_available():
34
  setup_eval_logging()
35
  OUTPUT_DIR = Path("./output/gradio")
36
  OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
 
37
  snapshot_download(repo_id="junxiliu/Meanaudio", local_dir="./weights",allow_patterns=["*.pt", "*.pth"] )
38
  current_model_state = {
39
  "net": None,
 
34
  setup_eval_logging()
35
  OUTPUT_DIR = Path("./output/gradio")
36
  OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
37
+ snapshot_download(repo_id="google/flan-t5-large")#, local_dir="./weights",allow_patterns=["*.pt", "*.pth"] )
38
  snapshot_download(repo_id="junxiliu/Meanaudio", local_dir="./weights",allow_patterns=["*.pt", "*.pth"] )
39
  current_model_state = {
40
  "net": None,