Update app.py

app.py

@@ -4,8 +4,12 @@ import tensorflow as tf
 
 import gradio as gr
 
+from transformers import AutoTokenizer, pipeline,  AutoModelForTokenClassification
+
+tokenizer = AutoTokenizer.from_pretrained("Battu007/1_Image_Classification")
 model = AutoModelForTokenClassification.from_pretrained("Battu007/1_Image_Classification")
 
+
 def predict_image(opened_image):
     img_array = tf.keras.utils.img_to_array(opened_image)
     img_array = tf.expand_dims(img_array, 0) #Convert image to one empty batch -> Model was trained on batches