Update app.py

app.py

@@ -49,7 +49,7 @@ text_to_speech_pipelines = {}
 # Initialize pipelines for other tasks
 visual_qa_pipeline = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa", use_auth_token=read_token)
 document_qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2", use_auth_token=read_token)
-image_classification_pipeline = pipeline("image-classification", model="facebook/detr-resnet-50", use_auth_token=read_token)
+image_classification_pipeline = pipeline("image-classification", model="facebook/deit-base-distilled-patch16-224", use_auth_token=read_token)
 object_detection_pipeline = pipeline("object-detection", model="facebook/detr-resnet-50", use_auth_token=read_token)
 video_classification_pipeline = pipeline("video-classification", model="facebook/timesformer-base-finetuned-k400", use_auth_token=read_token)
 summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn", use_auth_token=read_token)