Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-pipeline = pipeline("image-classification")
+pipeline = pipeline(task = "image-classification", model = "microsoft/beit-base-patch16-224-pt22k-ft22k")
 
 st.title("Predict the class of an image")
 
@@ -17,4 +17,6 @@ if file_name is not None:
 
   col2.header("Probabilities")
   for p in predictions:
-    col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
+    col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
+    
+