Update app.py

app.py

@@ -105,7 +105,7 @@ def main():
                     
                     # Make prediction
                     prediction = model.predict(np.expand_dims(processed_image, axis=0))
-                    class_names = ['Artificial', 'Nature']  # Replace with your actual class names
+                    class_names = ['10', '7', '7.5', '8', '8.5', '9', '9.2', '9.5', '9.7']
                     
                     # Get prediction and confidence
                     predicted_class = class_names[np.argmax(prediction)]