Update app.py

app.py

@@ -1,3 +1,4 @@
+import io
 import os
 import sys
 
@@ -85,7 +86,7 @@ async def search(query: str, k: int):
 
     results = []
     for idx in top_k_indices:
-        img_byte_arr = BytesIO()
+        img_byte_arr = io.BytesIO()
         images[idx].save(img_byte_arr, format='PNG')
         img_byte_arr = img_byte_arr.getvalue()
         img_base64 = base64.b64encode(img_byte_arr).decode('utf-8')