davidmasip commited on
Commit
2a6b030
·
1 Parent(s): 0336419

auth token

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,10 +1,16 @@
1
 
 
 
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
  RACISM_MODEL = "davidmasip/racism"
6
- racism_analysis_pipe = pipeline("text-classification",
7
- model=RACISM_MODEL, tokenizer=RACISM_MODEL, use_auth_token=True)
 
 
 
 
8
 
9
 
10
  def racism_analysis(text):
 
1
 
2
+ import os
3
+
4
  import gradio as gr
5
  from transformers import pipeline
6
 
7
  RACISM_MODEL = "davidmasip/racism"
8
+ racism_analysis_pipe = pipeline(
9
+ "text-classification",
10
+ model=RACISM_MODEL,
11
+ tokenizer=RACISM_MODEL,
12
+ use_auth_token=os.getenv("access_token")
13
+ )
14
 
15
 
16
  def racism_analysis(text):