Update app.py

app.py

@@ -9,6 +9,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 from sklearn.preprocessing import LabelEncoder
 from huggingface_hub import hf_hub_download
+import base64
 
 # Dataset loading function with caching
 @st.cache_data