Update app.py

app.py

@@ -65,7 +65,7 @@ def CTXGen(τ, g_num, length_range, model_name):
         while count < gen_num:
             new_seq = None
             if is_stopped:
-                return pd.DataFrame(), "output.csv"
+                return "output.csv", pd.DataFrame()
 
             if time.time() - start_time > 1200:
                 break
@@ -83,7 +83,7 @@ def CTXGen(τ, g_num, length_range, model_name):
 
             for i in range(length):
                 if is_stopped:
-                    return pd.DataFrame(), "output.csv"
+                    return "output.csv", pd.DataFrame()
 
                 _, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)
                 idx_seq = torch.tensor(idx_seq).unsqueeze(0).to(device)
@@ -143,10 +143,14 @@ def CTXGen(τ, g_num, length_range, model_name):
                     })
                     out.to_csv("output.csv", index=False, encoding='utf-8-sig')
                     count += 1
-                    yield out, "output.csv"
-    return out, "output.csv"
+                    yield "output.csv", out
+    return "output.csv", out
 
 with gr.Blocks() as demo:
+    gr.Markdown("🔗 **[Label Prediction](https://huggingface.co/spaces/oucgc1996/CTXGen_Label_Prediction)**")
+    gr.Markdown("🔗 **[Unconstrained Generation](https://huggingface.co/spaces/oucgc1996/CTXGen_Unconstrained_generation)**")
+    gr.Markdown("🔗 **[Conditional Generation](https://huggingface.co/spaces/oucgc1996/CTXGen_conditional_generation)**")
+    gr.Markdown("🔗 **[Optimization Generation](https://huggingface.co/spaces/oucgc1996/CTXGen_optimization_generation)**")
     gr.Markdown("# Conotoxin Unconstrained Generation")
     gr.Markdown("#### Input")
     gr.Markdown("✅**τ**: temperature factor controls the diversity of conotoxins generated. The higher the value, the higher the diversity")
@@ -161,12 +165,12 @@ with gr.Blocks() as demo:
     with gr.Row():
         start_button = gr.Button("Start Generation")
         stop_button = gr.Button("Stop Generation")
-    with gr.Row():
-        output_df = gr.DataFrame(label="Generated Conotoxins")
     with gr.Row():
         output_file = gr.File(label="Download generated conotoxins")
-    
-    start_button.click(CTXGen, inputs=[τ, g_num, length_range, model_name], outputs=[output_df, output_file])
+    with gr.Row():
+        output_df = gr.DataFrame(label="Generated Conotoxins")
+
+    start_button.click(CTXGen, inputs=[τ, g_num, length_range, model_name], outputs=[output_file, output_df])
     stop_button.click(stop_generation, outputs=None)
 
 demo.launch()