Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
-from transformers import pipeline, AutoModel
+from transformers import pipeline, AutoConfig, AutoModelForSequenceClassification
+
+config_path = "dogs.pkl"
+config = AutoConfig.from_pretrained(config_path)
+
+model = AutoModelForSequenceClassification.from_pretrained("ericxlima/DogsClassifierModel", config=config)
 
-model = AutoModel.from_pretrained("ericxlima/DogsClassifierModel")
- 
 dogs = { 
     'Zwergspitz Dog': [],
     'Bouledogue Français Dog': [],