Afroz Alam commited on
Commit
aab6dba
·
1 Parent(s): 991c817
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,8 +3,7 @@ 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-detector-single",use_auth_token=auth_token)
7
- pipeline_zh = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-single-chinese",use_auth_token=auth_token)
8
 
9
 
10
  def predict_en(text):
 
3
  from transformers import pipeline
4
 
5
  auth_token = os.environ.get("access_token")
6
+ pipeline_en = pipeline(task="text-classification", model="afroz14/demomode")
 
7
 
8
 
9
  def predict_en(text):