From 53621b39fa3934d8c13513ca02617dd87da094a5 Mon Sep 17 00:00:00 2001
From: weize <nanguage@yahoo.com>
Date: Sun, 11 Feb 2024 11:22:15 +0100
Subject: [PATCH] Update test_bioengine.py

---
 public/test_bioengine.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/public/test_bioengine.py b/public/test_bioengine.py
index 45ca931..4a51a18 100644
--- a/public/test_bioengine.py
+++ b/public/test_bioengine.py
@@ -10,7 +10,7 @@
 
 SERVER_URL = "https://ai.imjoy.io"
 
-image = await fetch_image('https://zenodo.org/api/records/6647683/files/sample_input_0.tif/content')
+image = await fetch_image('https://zenodo.org/api/records/6647674/files/sample_input_0.tif/content')
 
 server = await connect_to_server(
         {"server_url": SERVER_URL, "method_timeout": 3000}
@@ -18,7 +18,7 @@
 triton = await server.get_service("triton-client")
 
 # get model RDF
-#ret = await triton.execute(inputs=[{'inputs': None, "model_id": "powerful-chipmunk", 'return_rdf':True}],
+#ret = await triton.execute(inputs=[{'inputs': None, "model_id": "affable-shark", 'return_rdf':True}],
 #                           model_name="bioengine-model-runner",
 #                           serialization="imjoy",
 #                          )
@@ -27,7 +27,7 @@
 # run the model
 in_img = image[None, None, ...]
 
-ret = await triton.execute(inputs=[{"inputs":[in_img], "model_id": "powerful-chipmunk"}],
+ret = await triton.execute(inputs=[{"inputs":[in_img], "model_id": "affable-shark"}],
                            model_name="bioengine-model-runner",
                            serialization="imjoy",
                           )
@@ -41,4 +41,4 @@
 ax1.set_title('input image')
 ax2.imshow(mask[0][1])
 ax2.set_title('predicted mask')
-plt.show()
\ No newline at end of file
+plt.show()