Update app.py

app.py

@@ -6,7 +6,7 @@ def predict_image(img):
   prediction=model.predict(img_4d)[0]
   return {class_names[i]: float(prediction[i]) for i in range(4)}
 
-pipe = pipeline(task = 'image-classification', model = "microsoft/beit-base-patch16-224")
+pipe = pipeline(task = 'image-classification', model = "google/vit-base-patch16-224")
 
 image = gr.inputs.Image(shape=(180,180))
 label = gr.outputs.Label(num_top_classes=4)