Update app.py

app.py

@@ -1,9 +1,14 @@
 import gradio as gr
 from transformers import pipeline
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+
+extractor = AutoFeatureExtractor.from_pretrained("Rajaram1996/FacialEmoRecog")
+
+model_heh = AutoModelForImageClassification.from_pretrained("Rajaram1996/FacialEmoRecog")
 
 pipe = pipeline(task="image-classification", 
                  # model that can do 22k-category classification
-                 model="microsoft/beit-base-patch16-224-pt22k-ft22k")
+                 model=model_heh)
 gr.Interface.from_pipeline(pipe, 
                            title="22k Image Classification",
                            description="Object Recognition using Microsoft BEIT",