Update apps.py

apps.py

@@ -3,6 +3,7 @@ from esm.models.esm3 import ESM3
 from esm.sdk.api import ESM3InferenceClient, ESMProtein, GenerationConfig
 import spaces
 import os 
+import gradio as gr
 from gradio_molecule3d import Molecule3D
 
 
@@ -15,6 +16,14 @@ login(TOKEN)
 # This will download the model weights and instantiate the model on your machine.
 model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cuda")
 
+def read_mol(molpath):
+    with open(molpath, "r") as fp:
+        lines = fp.readlines()
+    mol = ""
+    for l in lines:
+        mol += l
+    return mol
+
 def molecule(input_pdb):
     mol = read_mol(input_pdb)
 
@@ -83,7 +92,7 @@ def prediction(prompt, temperature, do_structure, enable_roundtrip):
             protein = model.generate(protein, GenerationConfig(track="structure", num_steps=8))
             protein.to_pdb("./round_tripped.pdb")
             html1 = molecule("./round_tripped.pdb")
-            return seq, protein.sequemce, html, html1, "./generation.pdb", "./round_tripped.pdb"
+            return seq, protein.sequence, html, html1, "./generation.pdb", "./round_tripped.pdb"
         else:
             html1 = "<h3>Inverse folding and re-generation not enabled</h3>"
             f = open("./round_tripped.pdb", "w")
@@ -102,4 +111,4 @@ demo = gr.Interface(fn = prediction, inputs = [gr.Textbox(label="Masked protein
     <h3 align='center'>Support this space with a ⭐ on <a href='https://github.com/AstraBert/proteins-w-esm'>GitHub</a></h3>
     <h3 align='center'>Support Evolutionary Scale's ESM with a ⭐ on <a href='https://github.com/evolutionaryscale/esm'>GitHub</a></h3>""", examples = [["___________________________________________________DQATSLRILNNGHAFNVEFDDSQDKAVLKGGPLDGTYRLIQFHFHWGSLDGQGSEHTVDKKKYAAELHLVHWNTKYGDFGKAVQQPDGLAVLGIFLKVGSAKPGLQKVVDVLDSIKTKGKSADFTNFDPRGLLPESLDYWTYPGSLTTPP___________________________________________________________", 0.7, "No", "No"], ["__________________________________________________________AGQEEYSAMRDQYMRTGEGFLCVFAINNTKSFEDIHQYREQIKRVKDSDDVPMVLVGNKCDLAARTVESRQAQDLARSYGIPYIETSAKTRQGVEDAFYTLVRE___________________________", 0.2, "Yes", "No"], ["__________KTITLEVEPSDTIENVKAKIQDKEGIPPDQQRLIFAGKQLEDGRTLSDYNIQKESTLH________", 0.5, "Yes", "Yes"]], cache_examples=False)
 
-demo.launch()
+demo.launch()