beyond commited on
Commit
5120739
1 Parent(s): ac2870e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
-
5
- pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-qa-detector-distil",use_auth_token=access_token)
6
  # pipeline_en = pipeline_zh
7
  # pipeline_zh = pipeline(task="text2text-generation", model="beyond/genius-base-chinese")
8
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ auth_token = os.environ.get("access_token")
5
+ pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-qa-detector-distil",use_auth_token=auth_token)
6
  # pipeline_en = pipeline_zh
7
  # pipeline_zh = pipeline(task="text2text-generation", model="beyond/genius-base-chinese")
8