Update app.py

app.py

@@ -104,7 +104,8 @@ css = """
     footer {visibility: hidden}
 """
 
-with gr.Blocks(theme="huggingface", css=css) as demo:
+# 使用 Blocks API 创建界面
+with gr.Blocks(css=css) as demo:
     gr.Markdown("# 🌟 NSFW 图片分类器")
     gr.Markdown(description)
 
@@ -124,8 +125,8 @@ with gr.Blocks(theme="huggingface", css=css) as demo:
 
     submit_btn.click(predict, inputs=image_input, outputs=output_label)
 
-    # 启动 API 接口
-    gr.Interface(fn=api_predict, inputs=gr.Image(), outputs=gr.JSON(), api=True)
+    # 手动为 API 设置路由
+    demo.add_api_route("/api/predict", api_predict, methods=["POST"])
 
 # 启动 Web 界面
 demo.launch()