Update app.py

app.py

@@ -10,7 +10,7 @@ def classify_image(image):
 
         # Create the image classification pipeline
         img_class = pipeline(
-            "image-classification", model="AMfeta99/vit-base-oxford-brain-tumor"
+            "image-classification", model="AMfeta99/vit-base-oxford-brain-tumor_x-ray"
         )
 
         # Perform image classification