Update app.py

app.py

@@ -47,6 +47,21 @@ def predict_dog_breed(image):
     img_array = preprocess_image(image)
     predictions = model.predict(img_array)
     class_idx = np.argmax(predictions)  # Get index of the class with the highest probability
+    
+    # Ensure that class_idx is within the range of class_labels
+    if class_idx < len(class_labels):
+        breed = class_labels[class_idx]  # Map the index to the breed name
+        confidence = predictions[0][class_idx]  # Confidence score
+        return [breed, confidence]  # Return as a list
+    else:
+        return ["Error", "Invalid model output"]
+
+
+        
+def old_predict_dog_breed(image):
+    img_array = preprocess_image(image)
+    predictions = model.predict(img_array)
+    class_idx = np.argmax(predictions)  # Get index of the class with the highest probability
     breed = class_labels[class_idx]  # Map the index to the breed name
     confidence = predictions[0][class_idx]  # Confidence score
     return [breed, confidence]  # Return as a list