Add model

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

app.py

@@ -12,7 +12,7 @@ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png
 
 
 # Load your TensorFlow model
-# model = tf.keras.models.load_model("model.h5")
+model = tf.keras.models.load_model("model/kd_model")
 
 
 def preprocess_image(image, target_size=(224, 224)):
@@ -24,18 +24,17 @@ def preprocess_image(image, target_size=(224, 224)):
 col1, col2 = st.columns(2)
 
 if uploaded_file is not None:
-    image = Image.open(uploaded_file)
+    image = Image.open(uploaded_file).convert('RGB')
     col1.image(image, caption="Uploaded Image", use_column_width=True)
 
     if st.button("Classify"):
         preprocessed_image = preprocess_image(image)
-        # predictions = model.predict(preprocessed_image)
-        # top_prediction = np.argmax(predictions[0])
+        predictions = model.predict(preprocessed_image)
+        top_prediction = np.argmax(predictions[0])
 
-        # class_labels = get_class_labels()
-        # predicted_class = class_labels[top_prediction]
+        predicted_class = LABELS[top_prediction]
 
-        col2.write(f"Predicted class: ")  # {predicted_class}
+        col2.write(f"Predicted class: {predicted_class}")
 else:
     col1.write("Upload an image to see the classification")
     col2.write("Prediction will appear here")

model/kd_model/fingerprint.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:690937eedceddf7e2a78dee2089ec5f6a349d38376e98d8d04ba16160f640b70
+size 56

model/kd_model/keras_metadata.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bfa0da94b6ee8b5f2b48ac1b84b18dfc63125d9b407b6962ce0b73b5c3d9f7a
+size 39768

model/kd_model/saved_model.pb

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