Update app.py

app.py

@@ -135,10 +135,10 @@ def CTXGen(X1, X2, τ, g_num, length_range):
                             'ID':IDs,
                             'Generated_seq': generated_seqs_FINAL,
                             'Subtype': X1,
-                            'cls_probability': cls_probability_all, 
+                            'Subtype_probability': cls_probability_all, 
                             'Potency': X2, 
                             'Potency_probability': act_probability_all, 
-                            'random_seed': seed
+                            'Random_seed': seed
                         })
                         out.to_csv("output.csv", index=False, encoding='utf-8-sig')
                         count += 1
@@ -146,7 +146,7 @@ def CTXGen(X1, X2, τ, g_num, length_range):
     return out, "output.csv"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Conotoxin Generation")
+    gr.Markdown("# Conotoxin Conditional Generation")
     with gr.Row():
         X1 = gr.Dropdown(choices=['<α7>','<AChBP>','<α4β2>','<α3β4>','<Ca22>','<α3β2>', '<Na12>','<α9α10>','<K16>', '<α1β1γδ>', 
                                   '<K13>', '<α1BAR>', '<α1β1ε>', '<α1AAR>', '<GluN3A>', '<GluN2B>', '<α75HT3>', '<Na14>',