feat: add different pLMs as input

README.md

@@ -11,3 +11,5 @@ short_description: 'input: protein sequences; output: embeddings'
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+Minimal example Gradio app for inputting protein sequences and outputting protein LLM embeddings.

app.py

@@ -1,9 +1,16 @@
 import gradio as gr
 from embed import gen_embedding
 
-def generate_embeddings(sequences):
-    embeddings = gen_embedding(sequences, plm_model="esm1b")
+def generate_embeddings(sequences, plm_model):
+    embeddings = gen_embedding(sequences, plm_model=plm_model)
     return embeddings.tolist()
 
-demo = gr.Interface(fn=generate_embeddings, inputs="text", outputs="text")
+demo = gr.Interface(
+    fn=generate_embeddings,
+    inputs=[
+        "text", 
+        gr.Dropdown(choices=["esm1b", "esm2", "prott5", "prostt5"], label="PLM Model")
+    ],
+    outputs="text"
+)
 demo.launch()