small fix

README.md

@@ -8,4 +8,7 @@ pinned: false
 license: apache-2.0
 ---
 
+This app and code is the implementation of DockFormer described in https://www.biorxiv.org/content/10.1101/2024.11.25.625135v1.
+
+
 This code is built upon OpenFold. If you use or reference this code, please ensure you also cite the papers referenced on their GitHub page: https://github.com/aqlaboratory/openfold

inference_app.py

@@ -75,7 +75,7 @@ with gr.Blocks() as app:
     metrics = gr.JSON(label="Metrics")
     run_time = gr.Textbox(label="Runtime")
 
-    btn.click(predict, inputs=[input_sequence, input_ligand, input_msa, input_protein],
+    btn.click(predict, inputs=[input_sequence, input_ligand, input_msa, input_protein, model_variation],
               outputs=[out, metrics, run_time])
 
 app.launch()