ankit666 commited on
Commit
fcdd659
1 Parent(s): 4d30001

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -8,6 +8,14 @@ model_path = 'model__xception.h5'
8
  # Load the pre-trained model
9
  model = tf.keras.models.load_model(model_path)
10
 
 
 
 
 
 
 
 
 
11
  ## Define the predict function for Gradio
12
  def predict_gradio(image):
13
  # Preprocess the input image
 
8
  # Load the pre-trained model
9
  model = tf.keras.models.load_model(model_path)
10
 
11
+ #Defining the labels
12
+ labels = ['Aloevera','Amla','Amruthaballi','Arali','Astma_weed','Badipala','Balloon_Vine','Bamboo','Beans','Betel','Bhrami','Bringaraja','Caricature','Castor','Catharanthus',
13
+ 'Chakte','Chilly','Citron lime (herelikai)','Coffee','Common rue(naagdalli)','Coriender','Curry','Doddpathre','Drumstick','Ekka','Eucalyptus','Ganigale','Ganike',
14
+ 'Gasagase','Ginger','Globe Amarnath','Guava','Henna','Hibiscus','Honge','Insulin','Jackfruit','Jasmine','Kambajala','Kasambruga','Kohlrabi','Lantana','Lemon',
15
+ 'Lemongrass','Malabar_Nut','Malabar_Spinach','Mango','Marigold','Mint','Neem','Nelavembu','Nerale','Nooni','Onion','Padri','Palak(Spinach)','Papaya','Parijatha',
16
+ 'Pea','Pepper','Pomoegranate','Pumpkin','Raddish','Rose','Sampige','Sapota','Seethaashoka','Seethapala','Spinach1','Tamarind','Taro','Tecoma','Thumbe','Tomato',
17
+ 'Tulsi','Turmeric','ashoka','camphor','kamakasturi','kepala']
18
+
19
  ## Define the predict function for Gradio
20
  def predict_gradio(image):
21
  # Preprocess the input image