Update app.py

app.py

@@ -3,12 +3,18 @@ import os
 import random
 from src.model import simlarity_model as model
 from src.similarity.similarity import Similarity
+from PIL import Image
+import glob
 
 similarity = Similarity()
 models = similarity.get_models()
 
-def check(img_main, img_1, img_2, model_idx):    
-    result = similarity.check_similarity([img_main, img_1, img_2], models[model_idx])
+def check(img_main, model_idx):    
+    image_list = []
+    for filename in glob.glob('images/*.jpg'):
+        im=Image.open(filename)
+        image_list.append(im)
+    result = similarity.check_similarity([img_main] + image_list, models[model_idx])
     return result
 
 with gr.Blocks() as demo:
@@ -27,6 +33,6 @@ with gr.Blocks() as demo:
         ).style(grid=[2], height="auto")
 
     submit_btn = gr.Button('Check Similarity')
-    submit_btn.click(fn=check,inputs=[img_main, img_1, img_2, model], outputs=gallery)
+    submit_btn.click(fn=check,inputs=[img_main, model], outputs=gallery)
 
 demo.launch()