ThisUsernameAlreadyExistsAlreadyExists commited on
Commit
2bfde2c
·
1 Parent(s): db83cda

fix: gated model auth

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  import librosa
3
  import numpy as np
@@ -7,14 +8,17 @@ from transformers import (
7
  )
8
  import gradio as gr
9
 
 
10
  MODEL_ID = "aitf-komdigi/KomdigiITS-86M-DFK-DeepfakeAudioClassification"
11
 
12
  feature_extractor = AutoFeatureExtractor.from_pretrained(
13
- MODEL_ID
 
14
  )
15
 
16
  model = AutoModelForAudioClassification.from_pretrained(
17
- MODEL_ID
 
18
  )
19
 
20
  model.eval()
 
1
+ import os
2
  import torch
3
  import librosa
4
  import numpy as np
 
8
  )
9
  import gradio as gr
10
 
11
+ TOKEN = os.environ['HF_TOKEN']
12
  MODEL_ID = "aitf-komdigi/KomdigiITS-86M-DFK-DeepfakeAudioClassification"
13
 
14
  feature_extractor = AutoFeatureExtractor.from_pretrained(
15
+ MODEL_ID,
16
+ token=TOKEN,
17
  )
18
 
19
  model = AutoModelForAudioClassification.from_pretrained(
20
+ MODEL_ID,
21
+ token=TOKEN,
22
  )
23
 
24
  model.eval()