Deploy model to Hugging Face Space

app.py

@@ -0,0 +1,12 @@
+import gradio as gr
+from transformers import pipeline
+
+# 加载模型
+classifier = pipeline("sentiment-analysis", model="suredream/ml-cicd-demo")
+
+# 定义预测函数
+def predict(text):
+    return classifier(text)
+
+# 创建 Gradio 接口
+gr.Interface(fn=predict, inputs="text", outputs="label").launch()

requirements.txt

@@ -0,0 +1,5 @@
+transformers
+datasets
+wandb
+gradio
+pyyaml