Update app.py

app.py

@@ -38,10 +38,10 @@ def predict(image):
     features = extract_resnet_features(image)
     
     # Make prediction using Random Forest classifier
-    predicted_index = rf_classifier.predict([features])[0]
+    predicted_class = rf_classifier.predict([features])[0]
     
     # Decode predicted class using the class labels obtained from the Random Forest classifier
-    predicted_class = class_labels[predicted_index]
+    # predicted_class = class_labels[predicted_index]
     return predicted_class
 
 # Define Gradio interface