trial

.gitignore

@@ -1 +1,2 @@
-__pycache__/*
+__pycache__
+*.pyc

app.py

@@ -242,22 +242,11 @@ with gr.Blocks(theme='NoCrypt/miku', css=custom_css) as demo:
             run_button = gr.Button("🔍 Search")
         
     with gr.Column(elem_classes=["output-container"]):
-        # output_table = gr.DataFrame(
-        #     headers=node_attributes+['euclidean','cosine','dot_product'],
-        #     label="Nearest Phages in INPHARED"
-        # )
-        # output_table.download_button = True
         output_table = gr.DataFrame(
             headers=node_attributes+['euclidean','cosine','dot_product'],
-            label="Nearest Phages in INPHARED",
-            interactive=False,
-            wrap=True,
-            type="pandas",
-            elem_id="output_table",
-            value=None,
-            datatype="str",
-            download_button=True
+            label="Nearest Phages in INPHARED"
         )
+        output_table.download_button = True
 
     #gr.Markdown("https://www.biorxiv.org/content/10.1101/2024.12.17.627486v1")
     

esm_utils.py

@@ -28,7 +28,7 @@ class EsmEmbedding:
         
         with torch.no_grad():
             outputs = self.model(**tokens, output_hidden_states=True)
-            hidden = outputs.hidden_states[-1][0].detach().to(torch.float64)  # shape: [seq_len, hidden_dim]
+            hidden = outputs.hidden_states[-1][0].detach()  # shape: [seq_len, hidden_dim]
             
         mean_embedding = hidden[1:-1].mean(dim=0)  # mean over non-[CLS]/[EOS]
         cls_embedding = hidden[0]  # CLS token