chore: files added

nuffin · Oct 16, 2024 · 7dff30f · 7dff30f
1 parent 2b8e070
commit 7dff30f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/llmsearch/clients/__init__.py b/llmsearch/clients/__init__.py
diff --git a/llmsearch/clients/localai.py b/llmsearch/clients/localai.py
@@ -136,7 +136,7 @@ def embedding(self, model: str, text: str) -> Optional[List[float]]:
 # Example usage:
 if __name__ == "__main__":
     # Initialize the client
-    client = LocalAIClient(base_url="http://localhost:8080", model="your-model-name")
+    client = LocalAIClient(base_url="http://localhost:8080")
 
     # Example 1: Generating text
     prompt = "Tell me a story about a brave knight."

diff --git a/requirements.txt b/requirements.txt
@@ -14,3 +14,5 @@ alembic==1.13.3
 black==24.10.0
 Flask-SQLAlchemy==3.1.1
 bcrypt==4.2.0
+tensorflow==2.17.0
+transformers==4.45.2