Spaces:

suredream
/

ml-cicd-demo

Sleeping

suredream commited on May 23

Commit

319fe49

verified ·

1 Parent(s): 59352cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ classifier = pipeline("text-classification", model="suredream/ml-cicd-demo")
 # 定义预测函数
 def predict(text):
-    return classifier(text)
 # 创建 Gradio 接口
 gr.Interface(fn=predict, inputs="text", outputs="label").launch()

 # 定义预测函数
 def predict(text):
+    return classifier(text)[0]['label']
 # 创建 Gradio 接口
 gr.Interface(fn=predict, inputs="text", outputs="label").launch()