Update app.py

app.py

@@ -7,8 +7,14 @@ from tensorflow.keras.applications.resnet50 import preprocess_input
 from tensorflow.keras.models import load_model
 from transformers import pipeline
 
+
 model = tf.keras.models.load_model('model_2.h5')
 
+
+
+pipe = pipeline(task="image-classification",
+                # model for 22k-category classification
+                model)
 #function
 def example(image):
     image = image.reshape(-1, 256, 256, 3)