Update app.py

app.py

@@ -11,13 +11,10 @@ from matplotlib import font_manager
 import textwrap
 import os
 import tempfile
-import pandas as pd
 from huggingface_hub import hf_hub_download
 
 model_weight = hf_hub_download(repo_id="BAAI/bge-visualized", filename="Visualized_m3.pth")
 
-data = pd.read_csv("image_url.csv")
-
 # Load Thai font
 thai_font = font_manager.FontProperties(fname='./Sarabun-Regular.ttf')
 
@@ -42,7 +39,7 @@ def visualize_results(results):
 
     for i, res in enumerate(results):
         try:
-            image_url = data.iloc[int(res.payload["index"])]['image_url']
+            image_url = res.payload['image_url']
             img = Image.open(image_url) if os.path.exists(image_url) else Image.open(BytesIO(requests.get(image_url).content))
             name = res.payload['name']
             if len(name) > 30: