Get rid of default models on some tools (#236)

The models I hardcoded were outdated for certain tools so I removed them. Now the tools will use the model suggestions from the API instead so we're always up to date. I left the hardcoded model on `textToSpeech` since the suggested models in the API don't work for now. (See [this pr](huggingface/transformers#24952))
huggingface · Jul 27, 2023 · 89ea88a · 89ea88a
1 parent ed07ddb
commit 89ea88a
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 3 deletions.
diff --git a/packages/agents/src/tools/imageToText.ts b/packages/agents/src/tools/imageToText.ts
@@ -18,7 +18,6 @@ export const imageToTextTool: Tool = {
 			await inference.imageToText(
 				{
 					data,
-					model: "nlpconnect/vit-gpt2-image-captioning",
 				},
 				{ wait_for_model: true }
 			)

diff --git a/packages/agents/src/tools/speechToText.ts b/packages/agents/src/tools/speechToText.ts
@@ -18,7 +18,6 @@ export const speechToTextTool: Tool = {
 			await inference.automaticSpeechRecognition(
 				{
 					data,
-					model: "facebook/wav2vec2-large-960h-lv60-self",
 				},
 				{ wait_for_model: true }
 			)

diff --git a/packages/agents/src/tools/textToImage.ts b/packages/agents/src/tools/textToImage.ts
@@ -22,7 +22,6 @@ export const textToImageTool: Tool = {
 		return await inference.textToImage(
 			{
 				inputs: data,
-				model: "stabilityai/stable-diffusion-2",
 			},
 			{ wait_for_model: true }
 		);
-Original file line number
+Diff line change
@@ Expand Up / @@ -18,7 +18,6 @@ export const imageToTextTool: Tool = { @@
     			await inference.imageToText(
     				{
     					data,
-    					model: "nlpconnect/vit-gpt2-image-captioning",
     				},
     				{ wait_for_model: true }
     			)
@@ Expand Down @@