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 }
  );