Update app.py

app.py

@@ -3,51 +3,34 @@ import tensorflow as tf
 import numpy as np
 import gradio as gr
 import io
-import json
 
 # Load the model
 model_path = 'final_teath_classifier.h5'
 model = tf.keras.models.load_model(model_path)
 
-# Define preprocessing function
-
-
 # Define prediction function
 def predict_image(image):
-    # Save the image to a file-like object
-    image_bytes = io.BytesIO()
-    image.save(image_bytes, format="JPEG")
-
-    # Load the image from the file-like object
-    image = tf.keras.preprocessing.image.load_img(image_bytes, target_size=(256, 256))
-    image = tf.keras.preprocessing.image.img_to_array(image)
+    # Load and preprocess the image
+    image = np.array(image)/255
     image = np.expand_dims(image, axis=0)
 
-    # Make a prediction
-    prediction = model.predict(image)
+    # Make predictions
+    predictions = model.predict(image)
 
-    # Get the probability of being 'Clean' or 'Carries'
-    probabilities = tf.nn.softmax(prediction, axis=-1)
-    predicted_class_index = np.argmax(probabilities)
-    if predicted_class_index == 0:
-        predicted_label = "Clean"
-        predicted_probability = probabilities[0][0] * 100  # Convert to percentage
-    else:
-        predicted_label = "Carries"
-        predicted_probability = probabilities[0][1] * 100  # Convert to percentage
+    # Get the predicted class and probability
+    predicted_class_index = np.argmax(predictions)
+    predicted_label = "Clean" if predicted_class_index == 0 else "Carries"
+    predicted_probability = np.max(tf.nn.softmax(predictions))
 
     # Return the prediction result as a dictionary
-    return {"Predicted Label": predicted_label, "Evaluate the teeth": f"{predicted_probability:.2f}%"}
-
+    return {"Predicted Label": predicted_label, "Probability": f"{predicted_probability * 100:.2f}%"}
 
 # Create the interface
 input_interface = gr.Image(type="pil")
-output_interface = "json"
-
 iface = gr.Interface(
     fn=predict_image,
     inputs=input_interface,
-    outputs=output_interface)
+    outputs="json")
 
 # Launch the interface
 iface.launch(share=True)