Update app.py

app.py

@@ -11,7 +11,7 @@ feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16
 
 def classify_image(image):
     # Convert the PIL Image to a format compatible with the feature extractor
-    image_pneumonia = np.array(image)
+    image = np.array(image)
     # Preprocess the image and prepare it for the model
     inputs_pneumonia = feature_extractor(images=image, return_tensors="pt")
     # Make prediction