Update app.py

app.py

@@ -91,10 +91,8 @@ with gr.Blocks() as demo:
         query_button = gr.Button("查詢")
     
     with gr.Row():
-        with gr.Column():
-            output_box = gr.Textbox(label="查詢結果", interactive=False)
-        with gr.Column():
-            image_output = gr.Image(label="嫌疑犯圖片", visible=True, elem_id="image-small")
+        output_box = gr.Textbox(label="查詢結果", interactive=False)
+        image_output = gr.Image(label="嫌疑犯圖片", visible=True, elem_id="suspect-image")
     
     indirect_button = gr.Button("間接證據", visible=False)
     indirect_output = gr.Textbox(label="間接證據內容", interactive=False, visible=False)
@@ -117,13 +115,14 @@ with gr.Blocks() as demo:
         outputs=[input_box, indirect_button, indirect_output, output_box, image_output]
     )
 
-# Add custom CSS for image size
-css = """
-#image-small img {
-    max-width: 150px;
-    max-height: 150px;
+# Add custom CSS to adjust image size
+demo.add_css("""
+#suspect-image img {
+    max-width: 100px;
+    max-height: 100px;
+    object-fit: contain;
 }
-"""
+""")
 
 # Launch the Gradio app
-demo.launch(css=css)
+demo.launch()