Add HuggingFace icon to components

This commit is contained in:
anovazzi1 2024-02-28 17:54:08 -03:00
commit 0dc45c3c50
4 changed files with 6 additions and 0 deletions

View file

@ -9,6 +9,7 @@ class HuggingFaceEmbeddingsComponent(CustomComponent):
documentation = ( documentation = (
"https://python.langchain.com/docs/modules/data_connection/text_embedding/integrations/sentence_transformers" "https://python.langchain.com/docs/modules/data_connection/text_embedding/integrations/sentence_transformers"
) )
icon="HuggingFace"
def build_config(self): def build_config(self):
return { return {

View file

@ -9,6 +9,8 @@ class HuggingFaceInferenceAPIEmbeddingsComponent(CustomComponent):
display_name = "HuggingFaceInferenceAPIEmbeddings" display_name = "HuggingFaceInferenceAPIEmbeddings"
description = "HuggingFace sentence_transformers embedding models, API version." description = "HuggingFace sentence_transformers embedding models, API version."
documentation = "https://github.com/huggingface/text-embeddings-inference" documentation = "https://github.com/huggingface/text-embeddings-inference"
icon="HuggingFace"
def build_config(self): def build_config(self):
return { return {

View file

@ -8,6 +8,8 @@ from langflow import CustomComponent
class HuggingFaceEndpointsComponent(CustomComponent): class HuggingFaceEndpointsComponent(CustomComponent):
display_name: str = "Hugging Face Inference API" display_name: str = "Hugging Face Inference API"
description: str = "LLM model from Hugging Face Inference API." description: str = "LLM model from Hugging Face Inference API."
icon="HuggingFace"
def build_config(self): def build_config(self):
return { return {

View file

@ -274,6 +274,7 @@ export const nodeIconsLucide: iconsType = {
Google: GoogleIcon, Google: GoogleIcon,
HNLoader: HackerNewsIcon, HNLoader: HackerNewsIcon,
HuggingFaceHub: HuggingFaceIcon, HuggingFaceHub: HuggingFaceIcon,
HuggingFace: HuggingFaceIcon,
HuggingFaceEmbeddings: HuggingFaceIcon, HuggingFaceEmbeddings: HuggingFaceIcon,
IFixitLoader: IFixIcon, IFixitLoader: IFixIcon,
Meta: MetaIcon, Meta: MetaIcon,