Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 import pandas as pd
 from PIL import Image
 from transformers import pipeline
-from transformers import AutoImageProcessor, AutoModelForImageClassification
+
+pipe = pipeline("image-classification", model="raffaelsiregar/dog-breeds-classification", use_auth_token=True)
 
 def dog_classifier(dog_image):
     image = Image.fromarray(dog_image)
-    processor = AutoImageProcessor.from_pretrained("raffaelsiregar/dog-breeds-classification")
-    dog_classifier = AutoModelForImageClassification.from_pretrained("raffaelsiregar/dog-breeds-classification")
-    output = dog_classifier(image)
+    output = pipe(image)
     
     # creating pandas dataframe
     df = pd.DataFrame(output)