Atin Sakkeer Hussain commited on
Commit
71a8852
1 Parent(s): 4bae2e1

Add app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -21,6 +21,14 @@ args = {"model": "./ckpts/M2UGen/checkpoint.pth", "llama_type": "7B", "llama_dir
21
  "mert_path": "m-a-p/MERT-v1-330M", "vit_path": "google/vit-base-patch16-224", "vivit_path": "google/vivit-b-16x2-kinetics400",
22
  "music_decoder": "musicgen", "music_decoder_path": "facebook/musicgen-medium"}
23
 
 
 
 
 
 
 
 
 
24
  generated_audio_files = []
25
 
26
  llama_type = args.llama_type
 
21
  "mert_path": "m-a-p/MERT-v1-330M", "vit_path": "google/vit-base-patch16-224", "vivit_path": "google/vivit-b-16x2-kinetics400",
22
  "music_decoder": "musicgen", "music_decoder_path": "facebook/musicgen-medium"}
23
 
24
+ class dotdict(dict):
25
+ """dot.notation access to dictionary attributes"""
26
+ __getattr__ = dict.get
27
+ __setattr__ = dict.__setitem__
28
+ __delattr__ = dict.__delitem__
29
+
30
+ args = dotdict(args)
31
+
32
  generated_audio_files = []
33
 
34
  llama_type = args.llama_type