add ben2 (#1179)

this pr will add ben2 

model available at : https://huggingface.co/PramaLLC/BEN2

---------

Co-authored-by: Lucain <[email protected]>

Author: not-lain

packages/tasks/src/model-libraries-snippets.ts

@@ -83,6 +83,20 @@ function get_prompt_from_diffusers_model(model: ModelData): string | undefined {
 	}
 }
 
+export const ben2 = (model: ModelData): string[] => [
+	`import requests
+from PIL import Image
+from ben2 import AutoModel
+
+url = "https://upload.wikimedia.org/wikipedia/commons/b/b6/Felis_catus-cat_on_snow.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+
+model = AutoModel.from_pretrained("${model.id}")
+model.to("cuda").eval()
+foreground = model.inference(image)
+`,
+];
+
 export const bertopic = (model: ModelData): string[] => [
 	`from bertopic import BERTopic
 

packages/tasks/src/model-libraries.ts

@@ -109,6 +109,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		countDownloads: `path_extension:"pth"`,
 		snippets: snippets.audioseal,
 	},
+	ben2: {
+		prettyLabel: "BEN2",
+		repoName: "BEN2",
+		repoUrl: "https://github.com/PramaLLC/BEN2",
+		snippets: snippets.ben2,
+		filter: false,
+	},
 	bertopic: {
 		prettyLabel: "BERTopic",
 		repoName: "BERTopic",