Update app.py

app.py

@@ -111,7 +111,7 @@ def predict_bind(base_model_path,PEFT_model_path,input_seq):
     tokenizer = AutoTokenizer.from_pretrained(base_model_path) 
 
     # Tokenize the sequence
-    inputs = tokenizer(input_sequence, return_tensors="pt", truncation=True, max_length=1024, padding='max_length')
+    inputs = tokenizer(input_seq, return_tensors="pt", truncation=True, max_length=1024, padding='max_length')
 
     # Run the model
     with torch.no_grad():
@@ -380,7 +380,7 @@ with demo:
             """ Pick a base model and press **Finetune Pre-trained Model!"""
         )
         with gr.Row():   
-            with gr.Column(scale=0.5, variant="compact"):
+            with gr.Column(scale=5, variant="compact"):
                 base_model_name = gr.Dropdown(
                     choices=MODEL_OPTIONS,
                     value=MODEL_OPTIONS[0],
@@ -393,7 +393,7 @@ with demo:
                     label="PEFT Model Name",
                     interactive = True,
                 )
-            with gr.Column(scale=0.5, variant="compact"): 
+            with gr.Column(scale=5, variant="compact"): 
                     name = gr.Dropdown(
                         label="Choose a Sample Protein", 
                         value="Default protein", 
@@ -412,13 +412,13 @@ with demo:
                     placeholder="Paste your protein sequence here...",
                     interactive = True,
                 ) 
-        with gr.Column(variant="compact", scale = 2):
-                predict_btn = gr.Button(
-                    value="Predict binding site", 
-                    interactive=True, 
-                    variant="primary",
-                )
-                plot_struc_btn = gr.Button(value = "Plot ESMFold Predicted Structure ", variant="primary")
+            with gr.Column(variant="compact", scale = 2):
+                    predict_btn = gr.Button(
+                        value="Predict binding site", 
+                        interactive=True, 
+                        variant="primary",
+                    )
+                    plot_struc_btn = gr.Button(value = "Plot ESMFold Predicted Structure ", variant="primary")
         with gr.Row():  
             with gr.Column(variant="compact", scale = 5):
                 output_text = gr.Textbox(