Uhhy commited on
Commit
0c69cd4
·
verified ·
1 Parent(s): 7eac918

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ def load_model():
24
 
25
  processor_bytes = io.BytesIO()
26
  processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
27
- processor.tokenizer.save_pretrained(processor_bytes) # Save only the tokenizer
28
- processor.feature_extractor.save_pretrained(processor_bytes) # Save only the feature extractor
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
+ 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"