aapot commited on
Commit
55566c2
1 Parent(s): b53cf17

Temporarily load models with auth token

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. app.py +2 -2
README.md CHANGED
@@ -5,6 +5,7 @@ colorFrom: purple
5
  colorTo: blue
6
  sdk: gradio
7
  sdk_version: 3.3.1
 
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
5
  colorTo: blue
6
  sdk: gradio
7
  sdk_version: 3.3.1
8
+ python_version: 3.10.2
9
  app_file: app.py
10
  pinned: false
11
  license: apache-2.0
app.py CHANGED
@@ -27,9 +27,9 @@ placeholder_youtube_embedded_html = '''
27
 
28
 
29
  model_wt = YoutubeVideoSimilarityModel.from_pretrained(
30
- 'mozilla-foundation/youtube_video_similarity_model_wt', use_auth_token=True)
31
  model_nt = YoutubeVideoSimilarityModel.from_pretrained(
32
- 'mozilla-foundation/youtube_video_similarity_model_nt', use_auth_token=True)
33
  cross_encoder_model_name_or_path = model_wt.cross_encoder_model_name_or_path
34
 
35
 
 
27
 
28
 
29
  model_wt = YoutubeVideoSimilarityModel.from_pretrained(
30
+ 'mozilla-foundation/youtube_video_similarity_model_wt', use_auth_token=os.environ.get('HF_TOKEN'))
31
  model_nt = YoutubeVideoSimilarityModel.from_pretrained(
32
+ 'mozilla-foundation/youtube_video_similarity_model_nt', use_auth_token=os.environ.get('HF_TOKEN'))
33
  cross_encoder_model_name_or_path = model_wt.cross_encoder_model_name_or_path
34
 
35