diff --git a/docs/xinfer.md b/docs/xinfer.md index 8deb7cc..a72ab71 100644 --- a/docs/xinfer.md +++ b/docs/xinfer.md @@ -3,5 +3,5 @@ ::: xinfer.list_models -::: xinfer.get_model +::: xinfer.create_model diff --git a/nbs/demo.ipynb b/nbs/demo.ipynb index cf6fe98..6e1a793 100644 --- a/nbs/demo.ipynb +++ b/nbs/demo.ipynb @@ -18,7 +18,7 @@ "text/html": [ "
                             Available Models                             \n",
        "┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━┓\n",
-       "┃ Backend       Model ID                           Input/Output        ┃\n",
+       "┃ Backend       Model ID                           Input --> Output    ┃\n",
        "┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━┩\n",
        "│ transformers  Salesforce/blip2-opt-2.7b          image-text --> text │\n",
        "│ transformers  sashakunitsyn/vlrm-blip2-opt-2.7b  image-text --> text │\n",
@@ -29,7 +29,7 @@
       "text/plain": [
        "\u001b[3m                             Available Models                             \u001b[0m\n",
        "┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━┓\n",
-       "┃\u001b[1m \u001b[0m\u001b[1mBackend     \u001b[0m\u001b[1m \u001b[0m┃\u001b[1m \u001b[0m\u001b[1mModel ID                         \u001b[0m\u001b[1m \u001b[0m┃\u001b[1m \u001b[0m\u001b[1mInput/Output       \u001b[0m\u001b[1m \u001b[0m┃\n",
+       "┃\u001b[1m \u001b[0m\u001b[1mBackend     \u001b[0m\u001b[1m \u001b[0m┃\u001b[1m \u001b[0m\u001b[1mModel ID                         \u001b[0m\u001b[1m \u001b[0m┃\u001b[1m \u001b[0m\u001b[1mInput --> Output   \u001b[0m\u001b[1m \u001b[0m┃\n",
        "┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━┩\n",
        "│\u001b[36m \u001b[0m\u001b[36mtransformers\u001b[0m\u001b[36m \u001b[0m│\u001b[35m \u001b[0m\u001b[35mSalesforce/blip2-opt-2.7b        \u001b[0m\u001b[35m \u001b[0m│\u001b[32m \u001b[0m\u001b[32mimage-text --> text\u001b[0m\u001b[32m \u001b[0m│\n",
        "│\u001b[36m \u001b[0m\u001b[36mtransformers\u001b[0m\u001b[36m \u001b[0m│\u001b[35m \u001b[0m\u001b[35msashakunitsyn/vlrm-blip2-opt-2.7b\u001b[0m\u001b[35m \u001b[0m│\u001b[32m \u001b[0m\u001b[32mimage-text --> text\u001b[0m\u001b[32m \u001b[0m│\n",
@@ -71,7 +71,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
@@ -80,7 +80,7 @@
        "'An animated character with long hair and a serious expression is eating a large burger at a table, with other characters in the background.'"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
diff --git a/tests/test_xinfer.py b/tests/test_xinfer.py
index 0293acb..4f508eb 100644
--- a/tests/test_xinfer.py
+++ b/tests/test_xinfer.py
@@ -4,7 +4,7 @@
 
 import unittest
 
-from xinfer import get_model
+from xinfer import create_model
 
 
 class Testxinfer(unittest.TestCase):
diff --git a/xinfer/model_factory.py b/xinfer/model_factory.py
index d110790..78e52a4 100644
--- a/xinfer/model_factory.py
+++ b/xinfer/model_factory.py
@@ -38,7 +38,7 @@ def list_models():
     table = Table(title="Available Models")
     table.add_column("Backend", style="cyan")
     table.add_column("Model ID", style="magenta")
-    table.add_column("Input/Output", style="green")
+    table.add_column("Input --> Output", style="green")
 
     for model in ModelRegistry.list_models():
         table.add_row(