beyond commited on
Commit
31d4d2d
1 Parent(s): 6feceb5

Update app.py

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