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}
>
-
+ {datasetIcons[icon]}
{title}
@@ -159,7 +168,7 @@ export default function Datasets() {