Update README.md

README.md

@@ -19,6 +19,17 @@ datasets:
 - A2H0H0R1/plant-disease-new
 license: apache-2.0
 ---
+#Inference Pipeline
+--
+from transformers import AutoModelForImageClassification, AutoProcessor
+
+model = AutoModelForImageClassification.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
+processor = AutoProcessor.from_pretrained("ozair23/autotrain-w5nk2-rvmqx")
+
+def predict(image):
+    inputs = processor(images=image, return_tensors="pt")
+    outputs = model(**inputs)
+    return outputs
 
 # Model Trained Using AutoTrain