feat: add data type generic

examples/nextjs/src/app/page.tsx

@@ -8,7 +8,7 @@ export default function Home() {
   const [input, setInput] = useState("I love transformers!");
 
   const { data, isLoading, mutate } = useTransformers({
-    task: "sentiment-analysis",
+    task: "object-detection",
     options: {
       dtype: "q8",
     },

src/index.ts

@@ -1,8 +1,17 @@
 import { createBirpc } from "birpc";
 import { useEffect, useRef, useState } from "react";
-import type { PipelineType, ProgressInfo } from "@huggingface/transformers";
+import type {
+  AllTasks,
+  PipelineType,
+  ProgressInfo,
+} from "@huggingface/transformers";
 import type { BirpcReturn } from "birpc";
-import type { ClientFunctions, PipelineProps, ServerFunctions } from "./types";
+import type {
+  ClientFunctions,
+  PipelineProps,
+  ServerFunctions,
+  UnwrapPromise,
+} from "./types";
 
 export function useTransformers<T extends PipelineType>({
   task,
@@ -13,7 +22,7 @@ export function useTransformers<T extends PipelineType>({
 
   const [isReady, setIsReady] = useState(false);
   const [isLoading, setIsLoading] = useState(false);
-  const [data, setData] = useState<any>();
+  const [data, setData] = useState<UnwrapPromise<ReturnType<AllTasks[T]>>>();
   const [progressInfo, setProgressInfo] = useState<ProgressInfo>();
 
   const mutate = async (...data: any) => {

src/types.ts

@@ -1,5 +1,7 @@
 import type { PipelineType, ProgressInfo } from "@huggingface/transformers";
 
+export type UnwrapPromise<T> = T extends Promise<infer U> ? U : T;
+
 export interface PretrainedModelOptions {
   progress_callback?: (progress: any) => void;
   config?: any;