Uhhy commited on
Commit
e9cb163
Β·
verified Β·
1 Parent(s): 0c69cd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -24,8 +24,11 @@ def load_model():
24
 
25
  processor_bytes = io.BytesIO()
26
  processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
27
- processor.tokenizer.save_pretrained(processor_bytes, push_to_hub=False)
28
- processor.feature_extractor.save_pretrained(processor_bytes, push_to_hub=False)
 
 
 
29
  processor = MusicgenProcessor.from_pretrained(processor_bytes)
30
 
31
  title = "MusicGen Streaming"
 
24
 
25
  processor_bytes = io.BytesIO()
26
  processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
27
+
28
+ # Save the tokenizer and feature extractor using the `save_vocabulary` and `save_pretrained` methods respectively
29
+ processor.tokenizer.save_vocabulary(processor_bytes)
30
+ processor.feature_extractor.save_pretrained(processor_bytes)
31
+
32
  processor = MusicgenProcessor.from_pretrained(processor_bytes)
33
 
34
  title = "MusicGen Streaming"