pass model and tokenizer as input to pipeline and add batch_size

app.py

@@ -77,6 +77,13 @@ with gr.Blocks(title="ESM2 Protein Embeddings") as demo:
             value=list(MODELS.values())[0],
             label="Select Model"
         )
+        batch_size = gr.Slider(
+            minimum=1,
+            maximum=128,
+            step=1,
+            value=32,
+            label="Batch Size"
+        )
 
 
     current_key = [key for key, value in MODELS.items() if value == model_dropdown.value][0]
@@ -96,7 +103,7 @@ with gr.Blocks(title="ESM2 Protein Embeddings") as demo:
 
     submit_btn.click(
         fn=full_embedding_pipeline,
-        inputs=[input_files],
+        inputs=[input_files, model_to_use, tokenizer_to_use, batch_size],
         outputs=[download_output, status_output]
     )