Update app.py

app.py

@@ -6,10 +6,12 @@ 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")
+#tokenizer = AutoTokenizer.from_pretrained("Battu007/1_Image_Classification")
+#model = AutoModelForTokenClassification.from_pretrained("Battu007/1_Image_Classification")
 
+class_names = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
 
+model = tf.keras.models.load_model()
 
 def predict_image(opened_image):
     img_array = tf.keras.utils.img_to_array(opened_image)