Update app.py

app.py

@@ -1,8 +1,10 @@
-from transformers import ViTFeatureExtractor, ViTModel, AutoModelForSequenceClassification
+from transformers import ViTFeatureExtractor, ViTModel
 from PIL import Image
 import requests
 
-loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/chest_xray_pneumonia_detection")
+feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
+loaded_model = ViTModel.from_pretrained("runaksh/chest_xray_pneumonia_detection")
+inputs = feature_extractor(images=image, return_tensors="pt")
 
 pipe = pipeline('image-classification', model=model_name, device=0)