Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoModel
+
+model = AutoModel.from_pretrained("ericxlima/DogsClassifierModel")
  
 dogs = {
     'Zwergspitz Dog': [],
@@ -18,7 +20,8 @@ dogs = {
     'Golden Retriever': [],
     }
 
-pipeline = pipeline(model="ericxlima/DogsClassifierModel")
+pipeline = pipeline("image-classification", model=model)
+
 
 def predict(image):
   predictions = pipeline(image)