Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from PIL import Image
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
 def load_image(img):
     im=Image.open(img)
@@ -16,4 +17,10 @@ with st.sidebar:
 
 img = st.file_uploader("Choose a Image:")
 if img is not None:
-    st.write(img)
+    st.write(img)
+    extractor = AutoFeatureExtractor.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
+    model = AutoModelForImageClassification.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
+    pred=model.predict(img)
+    st.code(pred)
+    
+