Update app.py

app.py

@@ -13,7 +13,7 @@ theme = theme.Theme()
 
 
 # Cell 1: Image Classification Model
-image_pipeline = pipeline(task="image-classification", model="rocioadlc/TrashNet_ResNet152V2")
+image_pipeline = pipeline(task="image-classification", model="guillen/vit-basura-test1")
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)