jzq11111 commited on
Commit
819c6e0
·
verified ·
1 Parent(s): 4acbaea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -21,12 +21,16 @@ snapshot_download(repo_id="jzq11111/mooncast", local_dir='./resources/')
21
  from inference import Model
22
  import base64
23
 
24
- model = Model()
25
- model.generate_config.max_new_tokens = 50 * 50 # no more than 20s per turn
 
26
 
27
  @spaces.GPU(duration=120)
28
  def process_json_and_generate_audio(prompt_audio_role0_file, prompt_text_role0, prompt_audio_role1_file, prompt_text_role1, json_dialogue_input_str):
29
  try:
 
 
 
30
  print(json_dialogue_input_str, type(json_dialogue_input_str))
31
  print(prompt_audio_role0_file, prompt_text_role0, prompt_audio_role1_file, prompt_text_role1)
32
  # json_data = json.loads(json_dialogue_input_str)
 
21
  from inference import Model
22
  import base64
23
 
24
+ # model = Model()
25
+ # model.generate_config.max_new_tokens = 50 * 50 # no more than 20s per turn
26
+ model = None
27
 
28
  @spaces.GPU(duration=120)
29
  def process_json_and_generate_audio(prompt_audio_role0_file, prompt_text_role0, prompt_audio_role1_file, prompt_text_role1, json_dialogue_input_str):
30
  try:
31
+ if model is None:
32
+ model = Model()
33
+ model.generate_config.max_new_tokens = 50 * 50 # no more than 20s per turn
34
  print(json_dialogue_input_str, type(json_dialogue_input_str))
35
  print(prompt_audio_role0_file, prompt_text_role0, prompt_audio_role1_file, prompt_text_role1)
36
  # json_data = json.loads(json_dialogue_input_str)