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(