qichenhuang commited on
Commit
c96a20e
1 Parent(s): 523a234

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -11
app.py CHANGED
@@ -15,7 +15,8 @@ def generate_music(text):
15
  padding=True,
16
  return_tensors="pt",
17
  )
18
- audio_values = model.generate(**inputs, do_sample=True, guidance_scale=3, max_new_tokens=1350)
 
19
  sampling_rate = model.config.audio_encoder.sampling_rate
20
  scipy.io.wavfile.write("musicgen_out.wav", rate=sampling_rate, data=audio_values[0, 0].numpy())
21
  file_path= "musicgen_out.wav"
@@ -46,7 +47,7 @@ with iface as demo:
46
  generate_button.click(fn=generate_music, inputs=input_text, outputs=output_audio)
47
 
48
  # 设置网页背景
49
- demo.style.background_image = gr.Image.open("test.jpg")
50
  demo.style.backgroundRepeat = "no-repeat"
51
  demo.style.backgroundSize = "cover"
52
 
@@ -54,15 +55,6 @@ with iface as demo:
54
  with gr.Row():
55
  gr.Image("logo.jpg", width=100)
56
 
57
-
58
-
59
-
60
-
61
-
62
-
63
-
64
-
65
-
66
 
67
  # 启动界面
68
  demo.queue(max_size=4).launch(share = True)
 
15
  padding=True,
16
  return_tensors="pt",
17
  )
18
+ audio_values = model.generate(**inputs, do_sample=True, guidance_scale=3
19
+ , max_new_tokens=1350)
20
  sampling_rate = model.config.audio_encoder.sampling_rate
21
  scipy.io.wavfile.write("musicgen_out.wav", rate=sampling_rate, data=audio_values[0, 0].numpy())
22
  file_path= "musicgen_out.wav"
 
47
  generate_button.click(fn=generate_music, inputs=input_text, outputs=output_audio)
48
 
49
  # 设置网页背景
50
+ demo.style.background_image = gr.Image.open("background.png")
51
  demo.style.backgroundRepeat = "no-repeat"
52
  demo.style.backgroundSize = "cover"
53
 
 
55
  with gr.Row():
56
  gr.Image("logo.jpg", width=100)
57
 
 
 
 
 
 
 
 
 
 
58
 
59
  # 启动界面
60
  demo.queue(max_size=4).launch(share = True)