Add application file

Dockerfile

@@ -18,7 +18,7 @@ RUN pip install --no-cache-dir -r /code/requirements.txt
 RUN conda install pytorch::faiss-gpu --yes
 
 # Download foldseek
-RUN wget wget 'https://drive.usercontent.google.com/download?id=1B_9t3n_nlj8Y3Kpc_mMjtMdY0OPYa7Re&export=download&authuser=0' -O /tmp/foldseek
+RUN wget 'https://drive.usercontent.google.com/download?id=1B_9t3n_nlj8Y3Kpc_mMjtMdY0OPYa7Re&export=download&authuser=0' -O /tmp/foldseek
 RUN chmod +x /tmp/foldseek
 
 # Download ProTrek model

run.py

@@ -1,16 +1,39 @@
-import gradio as gr
+# import gradio as gr
+#
+#
+# def update(name):
+#     return f"Welcome to Gradio, {name}!"
+#
+#
+# with gr.Blocks() as demo:
+#     gr.Markdown("Start typing below and then click **Run** to see the output.")
+#     with gr.Row():
+#         inp = gr.Textbox(placeholder="What is your name?")
+#         out = gr.Textbox()
+#     btn = gr.Button("Run")
+#     btn.click(fn=update, inputs=inp, outputs=out)
+#
+# demo.launch()
+
+import sys
+root_dir = __file__.rsplit("/", 3)[0]
+if root_dir not in sys.path:
+    sys.path.append(root_dir)
 
+import gradio as gr
 
-def update(name):
-    return f"Welcome to Gradio, {name}!"
+from demo.modules.search import build_search_module
+from demo.modules.compute_score import build_score_computation
 
 
+# Build demo
 with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
-    with gr.Row():
-        inp = gr.Textbox(placeholder="What is your name?")
-        out = gr.Textbox()
-    btn = gr.Button("Run")
-    btn.click(fn=update, inputs=inp, outputs=out)
-
-demo.launch()
+    build_search_module()
+    build_score_computation()
+
+
+if __name__ == '__main__':
+    # args = get_args()
+
+    # Run demo
+    demo.launch()