data

app.py

@@ -92,7 +92,7 @@ def text_fn(taxon, name):
     text_embeds = torch.tensor(text_embeds)
     preds = model(text_embeds).sigmoid().unsqueeze(0).unsqueeze(0).detach().numpy()
     #return gr.Image(preds, label="Predicted Heatmap", visible=True)
-    return taxon + ": " + name + ": " + str(np.mean(preds))
+    return taxon + ": " + name + ": " + str(np.mean(preds)), preds
 
 def pred_fn(taxon, name):
     if taxon=="Class":
@@ -128,9 +128,9 @@ with gr.Blocks() as demo:
         submit_button = gr.Button("Run Model")
     
     with gr.Row():
-        pred = gr.Image(label="Predicted Heatmap", visible=False)
+        pred = gr.Image(label="Predicted Heatmap", visible=True)
     
-    check_button.click(text_fn, inputs=[inp, out], outputs=[text])
-    submit_button.click(pred_fn, inputs=[inp, out], outputs=[pred])
+    check_button.click(text_fn, inputs=[inp, out], outputs=[text, pred])
+    #submit_button.click(pred_fn, inputs=[inp, out], outputs=[pred])
     
 demo.launch()