Update app.py

app.py

@@ -4,6 +4,7 @@ import random
 import matplotlib.pyplot as plt
 import matplotlib.image as mpimg
 from embedding import model, docs_list, doc_embeddings, embedded_dict
+from preprocess import meme_attribute, meme_filename, meme_list
 import openai
 from openai import OpenAI
 import os
@@ -50,8 +51,8 @@ def create_meme(prompt):
     closest_text_idx = np.argmax(similarities)
     
     # Tải ảnh meme tương ứng
-    img_path = f'/kaggle/input/memedata/{embedded_dict[closest_text_idx]["filename"]}'
-    img = mpimg.imread(img_path)
+    img_path = meme_filename[closest_text_idx]
+    img = mpimg.imread(f'/kaggle/input/memedata/{img_path}')
     
     # Tạo khoảng trắng để hiển thị caption
     white_space_height = 100