updated app.py

app.py

@@ -7,8 +7,8 @@ from src.similarity.similarity import Similarity
 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, img_1, img_2, img_3, img_4, img_5, img_6, img_7, img_8, img_9, img_10):    
+    result = similarity.check_similarity([img_main, img_1, img_2, img_3, img_4, img_5, img_6, img_7, img_8, img_9, img_10], models[2])
     return result
 
 with gr.Blocks() as demo:
@@ -16,17 +16,35 @@ with gr.Blocks() as demo:
     img_main = gr.Text(label='Main Image', placeholder='https://myimage.jpg')
     
     gr.Markdown('Images to check')
+    # img_list = []
+    # for i in range(10):
+    #     img_list.append(gr.Text(label='{} st Image'.format(i), placeholder='https://myimage_1.jpg', i))
     img_1 = gr.Text(label='1st Image', placeholder='https://myimage_1.jpg')
+    print("img1: ", img_1)
     img_2 = gr.Text(label='2nd Image', placeholder='https://myimage_2.jpg')
+    img_3 = gr.Text(label='3st Image', placeholder='https://myimage_1.jpg')
+    img_4 = gr.Text(label='4nd Image', placeholder='https://myimage_2.jpg')
+    img_5 = gr.Text(label='5st Image', placeholder='https://myimage_1.jpg')
+    img_6 = gr.Text(label='6nd Image', placeholder='https://myimage_2.jpg')
+    img_7 = gr.Text(label='7st Image', placeholder='https://myimage_1.jpg')
+    img_8 = gr.Text(label='8nd Image', placeholder='https://myimage_2.jpg')
+    img_9 = gr.Text(label='9st Image', placeholder='https://myimage_1.jpg')
+    img_10 = gr.Text(label='10st Image', placeholder='https://myimage_2.jpg')
+    
+    # img_list = [img_1, img_2, img_3, img_4, img_5, img_6, img_7, img_8, img_9, img_10]
+    # for i in range(len(img_list)):
+    #     img
 
-    gr.Markdown('Choose the model')
-    model = gr.Dropdown([m.name for m in models], label='Model', type='index')
+    # gr.Markdown('Choose the model')
+    # model = gr.Dropdown([m.name for m in models], label='Model', type='index')
+    # print("model name:", model)
     
     gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+        ).style(grid=[10], height="auto")
 
+    output_metric = gr.Text(label='output metrics')
     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, img_1, img_2, img_3, img_4, img_5, img_6, img_7, img_8, img_9, img_10], outputs=output_metric, api_name="predict")
 
-demo.launch()    
+demo.launch(share=True)