Update app.py

app.py

@@ -4,6 +4,8 @@ import tensorflow as tf
 
 import gradio as gr
 
+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