diff --git a/frontend/components/Datasets.tsx b/frontend/components/Datasets.tsx index 2912d01..9a2eb91 100644 --- a/frontend/components/Datasets.tsx +++ b/frontend/components/Datasets.tsx @@ -21,6 +21,15 @@ import axios from "axios"; import { useQuery } from "react-query"; import { API_URL } from "@/app/config"; import { imagePrefix } from "@/app/config"; +import { FaFileAudio, FaFileAlt } from "react-icons/fa"; + +const datasetIcons: { [key: string]: React.ReactElement } = { + nmt: , + llm: , + asr: , + tts: , + xlit: , +}; interface FeatureProps { title: string; @@ -58,7 +67,7 @@ const Feature = ({ title, icon }: FeatureProps) => { bg={"gray.100"} mb={1} > - Icon + {datasetIcons[icon]} {title} @@ -159,7 +168,7 @@ export default function Datasets() {