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")