Update app.py

app.py

@@ -5,15 +5,15 @@ from transformers import pipeline, ConvNextForImageClassification, ConvNextFeatu
 from PIL import Image
 
 #load the models
-convnext_model = ConvNextForImageClassification.from_pretrained('convnext-rice')
+convnext_model = ConvNextForImageClassification.from_pretrained('convnext')
 convnext_feature_extractor = ConvNextFeatureExtractor.from_pretrained('facebook/convnext-tiny-224')
 convnext_clf = pipeline("image-classification", model = convnext_model, feature_extractor = convnext_feature_extractor)
 
-vit_model = ViTForImageClassification.from_pretrained('vit-rice')
+vit_model = ViTForImageClassification.from_pretrained('vit')
 vit_feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
 vit_clf = pipeline("image-classification", model = vit_model, feature_extractor = vit_feature_extractor)
 
-resnet_model = ResNetForImageClassification.from_pretrained('resnet-rice')
+resnet_model = ResNetForImageClassification.from_pretrained('resnet')
 resnet_feature_extractor = AutoFeatureExtractor.from_pretrained('microsoft/resnet-50')
 resnet_clf = pipeline("image-classification", model = resnet_model, feature_extractor = resnet_feature_extractor)