Update app.py

app.py

@@ -1,5 +1,22 @@
 import gradio as gr
 from transformers import pipeline
+from transformers import TFBertForSequenceClassification
+import tensorflow as tf
+
+data_dir = "C:/Users/Evanslee/Desktop/data"
+
+img_height,img_width=180,180
+batch_size=32
+train_ds = tf.keras.preprocessing.image_dataset_from_directory(
+  data_dir,
+  validation_split=0.2,
+  subset="training",
+  seed=123,
+  image_size=(img_height, img_width),
+  batch_size=batch_size)
+  
+class_names = train_ds.class_names
+print(class_names)
 
 def predict_image(img):
   img_4d=img.reshape(-1,180,180,3)