fix torch load

app.py

@@ -46,7 +46,7 @@ model.load_state_dict(state['model'], strict=True)
 model.eval()
 ling_disc.eval()
 
-state = torch.load(args.sem_ckpt)
+state = torch.load(args.sem_ckpt, map_location=torch.device('cpu'))
 sem_emb.load_state_dict(state['model'], strict=True)
 sem_emb.eval()
 
@@ -267,11 +267,17 @@ body {
   --input-background: --neutral-50;
   }
 
-.separator {
+.top-separator {
   width: 100%;
   height: 3px; /* Adjust the height for boldness */
   background-color: #000; /* Adjust the color as needed */
-  margin: 20px 0; /* Adjust the margin as needed */
+  margin-top: 20px; /* Adjust the margin as needed */
+}
+.bottom-separator {
+  width: 100%;
+  height: 3px; /* Adjust the height for boldness */
+  background-color: #000; /* Adjust the color as needed */
+  margin-bottom: 20px; /* Adjust the margin as needed */
 }
 """
 
@@ -286,7 +292,7 @@ with gr.Blocks(
     with gr.Accordion("🚀 Quick Start Guide", open=False, elem_id='guide'):
         gr.Markdown(guide)
 
-    with gr.Group(elem_classes='separator'):
+    with gr.Group(elem_classes='top-separator'):
         pass
     with gr.Group(elem_id='mode'):
         mode = gr.Radio(
@@ -311,7 +317,7 @@ with gr.Blocks(
         with gr.Column():
             sent2 = gr.Textbox(label='Generated text')
     interpolation = gr.Textbox(label='Quality control interpolation', visible=False, lines=5)
-    with gr.Group(elem_classes='separator'):
+    with gr.Group(elem_classes='bottom-separator'):
         pass
     #####################
     with gr.Row():
@@ -329,16 +335,16 @@ with gr.Blocks(
     #####################
     generate_btn = gr.Button("Generate", variant='primary', visible=False)
     with gr.Accordion("Tools to assist in the setting of linguistic indices...", open=False, visible=False) as ling_tools:
-        with gr.Row():
-            estimate_tgt_btn = gr.Button("Estimate linguistic indices of this sentence", visible=False)
-            sent_ling_est = gr.Textbox(label='Text to estimate linguistic indices', scale=2, visible=False)
-        estimate_src_btn = gr.Button("Estimate linguistic indices of source sentence", visible=False)
-        # rand_btn = gr.Button("Random target")
         rand_ex_btn = gr.Button("Random target", size='lg', visible=False)
-        copy_btn = gr.Button("Copy linguistic indices of source to target", size='sm', visible=False)
+        with gr.Row():
+            estimate_src_btn = gr.Button("Estimate linguistic indices of source sentence", visible=False)
+            copy_btn = gr.Button("Copy linguistic indices of source to target", size='lg', visible=False)
         with gr.Row():
             sub_btn = gr.Button('Subtract \u03B5 from target linguistic indices', visible=False)
             add_btn = gr.Button('Add \u03B5 to target linguistic indices', visible=False)
+        with gr.Row():
+            estimate_tgt_btn = gr.Button("Estimate linguistic indices of this sentence →", visible=False)
+            sent_ling_est = gr.Textbox(label='Text to estimate linguistic indices', scale=2, visible=False, container=False)
     ling.render()
     #####################