ajanz commited on
Commit
6831777
1 Parent(s): 0d77014

added missing authorisation token

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,10 +1,14 @@
1
  import gradio as gr
2
  import datasets
3
  import faiss
 
4
 
5
  from transformers import pipeline
6
 
7
 
 
 
 
8
  sample_text = """Europejscy astronomowie odkryli planetę
9
  pozasłoneczną pochodzącą spoza naszej galaktyki, czyli
10
  [unused0] Drogi Mlecznej [unused1]. Obserwacji dokonali
@@ -28,7 +32,7 @@ def load_index(index_data: str = "clarin-knext/entity-linking-index"):
28
 
29
 
30
  def load_model(model_name: str = "clarin-knext/entity-linking-encoder"):
31
- pipe = pipeline("feature-extraction", model=model_name)
32
  return pipe
33
 
34
 
 
1
  import gradio as gr
2
  import datasets
3
  import faiss
4
+ import os
5
 
6
  from transformers import pipeline
7
 
8
 
9
+ auth_token = os.environ.get("CLARIN_KNEXT")
10
+
11
+
12
  sample_text = """Europejscy astronomowie odkryli planetę
13
  pozasłoneczną pochodzącą spoza naszej galaktyki, czyli
14
  [unused0] Drogi Mlecznej [unused1]. Obserwacji dokonali
 
32
 
33
 
34
  def load_model(model_name: str = "clarin-knext/entity-linking-encoder"):
35
+ pipe = pipeline("feature-extraction", model=model_name, use_auth_token=auth_token)
36
  return pipe
37
 
38