fix mobile

app.py

@@ -1,5 +1,5 @@
 import time
-from itertools import count, islice
+from itertools import islice
 from functools import partial
 from typing import Iterable, Iterator, TypeVar
 
@@ -140,10 +140,10 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column(scale=4, min_width=0):
                 pass
-            with gr.Column(scale=9):
-                search_bar = gr.Textbox(max_lines=1, placeholder="Search datasets, get infinite results", show_label=False, container=False)
-            with gr.Column(min_width=64):
-                search_button = gr.Button("🔍", variant="primary")
+            with gr.Column(scale=10):
+                with gr.Row():
+                    search_bar = gr.Textbox(max_lines=1, placeholder="Search datasets, get infinite results", show_label=False, container=False, scale=9)
+                    search_button = gr.Button("🔍", variant="primary", scale=1)
             with gr.Column(scale=4, min_width=0):
                 pass
         with gr.Row():