StevenChen16 commited on
Commit
1fb173b
β€’
1 Parent(s): 31ba7c5

second commit

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +2 -0
app.py CHANGED
@@ -257,10 +257,10 @@ if __name__ == "__main__":
257
  parser = ArgumentParser()
258
  parser.add_argument("--host", type=str, default="0.0.0.0")
259
  parser.add_argument("--port", type=int, default="7860")
260
- parser.add_argument("--flow-path", type=str, default="./glm-4-voice-decoder")
261
- parser.add_argument("--model-path", type=str, default="./weights")
262
- parser.add_argument("--tokenizer-path", type=str, default="./glm-4-voice-tokenizer")
263
- parser.add_argument("--whisper_model", type=str, default="/root/autodl-tmp/whisper/base")
264
  parser.add_argument("--share", action='store_true')
265
  args = parser.parse_args()
266
 
 
257
  parser = ArgumentParser()
258
  parser.add_argument("--host", type=str, default="0.0.0.0")
259
  parser.add_argument("--port", type=int, default="7860")
260
+ parser.add_argument("--flow-path", type=str, default="THUDM/glm-4-voice-decoder")
261
+ parser.add_argument("--model-path", type=str, default="THUDM/glm-4-voice-9b")
262
+ parser.add_argument("--tokenizer-path", type=str, default="THUDM/glm-4-voice-tokenizer")
263
+ parser.add_argument("--whisper_model", type=str, default="base")
264
  parser.add_argument("--share", action='store_true')
265
  args = parser.parse_args()
266
 
requirements.txt CHANGED
@@ -36,3 +36,5 @@ torch==2.3.0
36
  torchaudio==2.3.0
37
  langchain
38
  whisper
 
 
 
36
  torchaudio==2.3.0
37
  langchain
38
  whisper
39
+ langchain-community
40
+ langchain-huggingface