Update app.py

app.py

@@ -2,21 +2,3 @@ from transformers import pipeline
 
 vision_classifier = pipeline(task="image-classification")
 result = vision_classifier(images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg")
-
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
-import torch
-from datasets import load_dataset
-
-dataset = load_dataset("huggingface/cats-image")
-image = dataset["test"]["image"][0]
-
-feature_extractor = AutoFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
-model = AutoModelForImageClassification.from_pretrained("google/vit-base-patch16-224")
-
-inputs = feature_extractor(image, return_tensors="pt")
-
-with torch.no_grad():
-    logits = model(**inputs).logits
-
-predicted_label = logits.argmax(-1).item()
-print(model.config.id2label[predicted_label])