Hugging Face's logo

Spaces:
suredream
/
ml-cicd-demo
Sleeping

App Files Community
suredream commited on
Commit
59352cb
·
verified ·
1 Parent(s): 7035188

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  # 加载模型
5
- classifier = pipeline("sentiment-analysis", model="suredream/ml-cicd-demo")
6
 
7
  # 定义预测函数
8
  def predict(text):
 
2
  from transformers import pipeline
3
 
4
  # 加载模型
5
+ classifier = pipeline("text-classification", model="suredream/ml-cicd-demo")
6
 
7
  # 定义预测函数
8
  def predict(text):