Use hub for model.

.ipynb_checkpoints/app-checkpoint.py

@@ -22,15 +22,23 @@ print(os.getcwd())
 data_dir = Path('.')
 data = load_data(data_dir, 'data.pkl')
 
+from huggingface_hub import hf_hub_download
+
+model_cache_path = hf_hub_download(repo_id="psistolar/musicautobot-fine1", filename="model.pth")
+
+
+
 # Default config options
 config = default_config()
 config['encode_position'] = True
 
+
+
 # Load our fine-tuned model
 learner = music_model_learner(
     data, 
-    config=config.copy(),
-    pretrained_path='model.pth'
+    config=config.copy(), 
+    pretrained_path=model_cache_path
 )
 
 

app.py

@@ -22,15 +22,23 @@ print(os.getcwd())
 data_dir = Path('.')
 data = load_data(data_dir, 'data.pkl')
 
+from huggingface_hub import hf_hub_download
+
+model_cache_path = hf_hub_download(repo_id="psistolar/musicautobot-fine1", filename="model.pth")
+
+
+
 # Default config options
 config = default_config()
 config['encode_position'] = True
 
+
+
 # Load our fine-tuned model
 learner = music_model_learner(
     data, 
-    config=config.copy(),
-    pretrained_path='model.pth'
+    config=config.copy(), 
+    pretrained_path=model_cache_path
 )