Update app.py

app.py

@@ -1,22 +1,26 @@
 import gradio as gr
+from transformers import pipeline, AutoTokenizer
 
-# 定义第一个接口
-def greet(name):
-    return "Hello " + name + "!!"
+# 加载模型和分词器
+model_name = "bert-base-uncased"
+model = pipeline("text-classification", model=model_name, tokenizer=model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-iface_greet = gr.Interface(fn=greet, inputs="text", outputs="text", title="Greeting")
+# 定义 Gradio 接口
+def classify_text(text):
+    # 对输入文本进行分词
+    inputs = tokenizer(text, return_tensors="pt")
+    # 使用模型进行预测
+    outputs = model(inputs)
+    # 获取预测结果
+    label = outputs[0]["label"]
+    score = outputs[0]["score"]
+    return f"Label: {label}, Score: {score:.2f}"
 
-# 定义第二个接口
-def count_numbers(n):
-    return len(str(n))
+iface = gr.Interface(
+    fn=classify_text,
+    inputs=gr.inputs.Textbox(),
+    outputs=gr.outputs.HTML(),
+)
 
-iface_count = gr.Interface(fn=count_numbers, inputs="number", outputs="number", title="Count Numbers")
-
-# 将两个接口添加到接口组中
-iface_group = gr.InterfaceGroup([iface_greet, iface_count], title="My Interfaces")
-
-# 为第一个接口添加一个按钮，点击该按钮将切换到第二个接口
-iface_greet.add_button("Count Numbers", interface=iface_count)
-
-# 启动接口组
-iface_group.launch()
+iface.launch()