Update app.py

app.py

@@ -26,16 +26,6 @@ CLASS_LABELS = {
 }
 
 def predict_image(image):
-    """
-    Predict the class of a given image.
-
-    Args:
-        image (PIL.Image.Image): Input image.
-
-    Returns:
-        str: Predicted class description.
-        float: Confidence score of the prediction.
-    """
     # Preprocess the image
     img_array = img_to_array(image)
     img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
@@ -50,7 +40,7 @@ def predict_image(image):
 
 # Streamlit app
 st.title("Driver Distraction Detection")
-st.markdown("Team18 Image Project : Sayandip Bhattacharyya, Purnendu Rudrapal, Sridatta Das, Sidharta Karjee")
+st.markdown("Team18 Image Project : Sayandip Bhattacharyya, Purnendu Rudrapal, Sridatta Das, Sidhartha Karjee")
 
 # File upload
 uploaded_file = st.file_uploader("Choose an image file", type=["jpg", "jpeg", "png"])