更新 README.md

2024-05-27 14:21:12 +08:00 · 2024-05-27 14:21:12 +08:00 · 8726b97d8e
parent 30ca939316
commit 8726b97d8e
1 changed files with 1 additions and 0 deletions
--- 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")