add model v5

app.py

@@ -32,17 +32,17 @@ def main():
         st.write('Prediction : {}'.format(result))
         st.write('Confidence : {}%'.format(confidence))
 
+imgSize = 256
 
 def predict_class(image):
     with st.spinner('Loading Model...'):
         classifier_model = keras.models.load_model(
-            r'plant_model_v1-beta.h5', compile=False)
+            r'plant_model_v5-beta.h5', compile=False)
 
-    shape = ((200, 200, 3))
-    # ye bhi kaam kar raha he
+    shape = ((imgSize, imgSize, 3))
     model = keras.Sequential(
         [hub.KerasLayer(classifier_model, input_shape=shape)])
-    test_image = image.resize((200, 200))
+    test_image = image.resize((imgSize, imgSize))
     test_image = keras.preprocessing.image.img_to_array(test_image)
     test_image /= 220.0
     test_image = np.expand_dims(test_image, axis=0)

plant_model_v5-beta.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:355bb4d8fdca1f3674a3f3c592353471fdd340ddc3c0ec211346a7de203c2598
+size 15806664