From 8726b97d8e2238e95c3d64f99a023b83aebc1a0a Mon Sep 17 00:00:00 2001 From: ailab Date: Mon, 27 May 2024 14:21:12 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20README.md?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 7b237cb..4e648eb 100644 --- a/README.md +++ b/README.md @@ -54,6 +54,7 @@ print(model.config.id2label[predicted_label]) ``` ```python +from transformers import pipeline import gradio as gr pipe = pipeline("image-classification", model="ailab/resnet-dogcat") gr.Interface.from_pipeline(pipe).launch(server_name="0.0.0.0")