Update README.md

README.md

@@ -50,7 +50,7 @@ The simplest way to perform inference is using the pipeline interface
 from transformers import pipeline
 image_url = "https://fema-cap-imagery.s3.amazonaws.com/Images/CAP_-_Flooding_Spring_2023/Source/IAWG_23-B-5061/A0005/D75_0793_DxO_PL6_P.jpg"
 
-pipe = pipeline(model="MITLL/LADI-v2-classifier-large")
+pipe = pipeline(model="MITLL/LADI-v2-classifier-small")
 print(pipe(image_url))
 ```
 
@@ -76,8 +76,8 @@ image_url = "https://fema-cap-imagery.s3.amazonaws.com/Images/CAP_-_Flooding_Spr
 img_data = requests.get(image_url).content
 img = Image.open(BytesIO(img_data))
 
-processor = AutoImageProcessor.from_pretrained("MITLL/LADI-v2-classifier-large")
-model = AutoModelForImageClassification.from_pretrained("MITLL/LADI-v2-classifier-large")
+processor = AutoImageProcessor.from_pretrained("MITLL/LADI-v2-classifier-small")
+model = AutoModelForImageClassification.from_pretrained("MITLL/LADI-v2-classifier-small")
 
 inputs = processor(img, return_tensors="pt")