Update app.py

app.py

@@ -89,7 +89,7 @@ def process_sequence(sequence, domain_bounds, n):
 
     df = pd.DataFrame({
         'Original Residue': original_residues,
-        'Predicted Residues (in order of decreasing likelihood)': mutations,
+        'Predicted Residues': mutations,
         'Position': positions
     })
             
@@ -98,17 +98,21 @@ def process_sequence(sequence, domain_bounds, n):
 demo = gr.Interface(
     fn=process_sequence,
     inputs=[
-        "text",
+        gr.Textbox(label="Sequence", placeholder="Enter the protein sequence here"),
         gr.Dataframe(
             headers=["start", "end"],
             datatype=["number", "number"],
             row_count=(1, "fixed"),
             col_count=(2, "fixed"),
+            label="Domain Bounds"
         ),
-        gr.Dropdown([i for i in range(1, 21)]),  # Dropdown with numbers from 1 to 20 as integers
+        gr.Dropdown([i for i in range(1, 21)], label="Top N Tokens"),  
     ],
-    outputs=["dataframe", "image"],
-    description="Choose a number between 1-20 to predict n tokens for each position. Choose the start and end index of the domain of interest (indexing starts at 1).",
+     outputs=[
+        gr.Dataframe(label="Predicted Tokens (in order of decreasing likelihood)"),
+        gr.Image(type="pil", label="Heatmap"),
+    ],
+    description="Choose a number from the dropdown to predict N tokens for each position. Choose the start and end index of the domain of interest (indexing starts at 1).",
 )
 
 if __name__ == "__main__":