mrfakename commited on
Commit
db455db
·
verified ·
1 Parent(s): 7634e1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,8 +1,11 @@
1
  from sonique import get_pretrained_model
2
  from sonique.interface.gradio import create_ui
3
  import json
4
-
5
  import torch
 
 
 
6
 
7
  interface = create_ui(
8
  model_config_path = str(cached_path('https://raw.githubusercontent.com/zxxwxyyy/sonique/refs/heads/main/best_model.json')),
 
1
  from sonique import get_pretrained_model
2
  from sonique.interface.gradio import create_ui
3
  import json
4
+ from huggingface_hub import login
5
  import torch
6
+ import os
7
+
8
+ login(token=os.getenv('HF_TOKEN'))
9
 
10
  interface = create_ui(
11
  model_config_path = str(cached_path('https://raw.githubusercontent.com/zxxwxyyy/sonique/refs/heads/main/best_model.json')),