dunktra commited on
Commit
c8e7c5f
·
verified ·
1 Parent(s): 3747e18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from Gradio_UI import GradioUI
12
  search_tool = DuckDuckGoSearchTool()
13
 
14
  # Load the MusicGen model
15
- pipe = pipeline("text-to-audio", model="facebook/musicgen-large")
16
 
17
  # Custom tool for AI-assisted music generation
18
  @tool
 
12
  search_tool = DuckDuckGoSearchTool()
13
 
14
  # Load the MusicGen model
15
+ pipe = pipeline("text-to-audio", model="facebook/musicgen-small")
16
 
17
  # Custom tool for AI-assisted music generation
18
  @tool