Update app.py

app.py

@@ -69,8 +69,8 @@ def caption_to_image(query_text: str, top_k: int) -> pd.DataFrame:
 
 def omni_to_image(
     query_text: str,
-    styles: list[int],
-    genres: list[int],
+    styles: list[str],
+    genres: list[str],
     tags: list[str],
     top_k: int,
 ) -> pd.DataFrame:
@@ -137,11 +137,11 @@ def build_demo() -> gr.Blocks:
             omni_topk = gr.Slider(3, 20, value=10, step=1, label="Top-K")
             omni_df = gr.DataFrame(headers=["Image Path", "Artist", "Style", "Genre", "Caption", "Tags"], datatype=["str", "str", "str", "str", "str", "str"], label="Результаты (номер сэмпла соответствует номеру в датасете)")
             omni_run = gr.Button("Объединить запрос")
-            # omni_run.click(
-            #     omni_to_image,
-            #     inputs=[query_box, [], [], [], omni_topk],
-            #     outputs=omni_df,
-            # )
+            omni_run.click(
+                omni_to_image,
+                inputs=[query_box, style_select, genre_select, tag_select, omni_topk],
+                outputs=omni_df,
+            )
 
     return demo