26 lines
726 B
TypeScript
26 lines
726 B
TypeScript
import { useState } from "react";
|
|
import { searchHfModels } from "./useGetDownloadedModels";
|
|
import { SearchModelParamHf } from "@/_models/hf/SearchModelParam.hf";
|
|
import { Product } from "@/_models/Product";
|
|
|
|
export default function useGetHuggingFaceModel() {
|
|
const [modelList, setModelList] = useState<Product[]>([]);
|
|
|
|
const getHuggingFaceModel = async (owner?: string) => {
|
|
if (!owner) {
|
|
setModelList([]);
|
|
return;
|
|
}
|
|
|
|
const searchParams: SearchModelParamHf = {
|
|
search: { owner },
|
|
limit: 5,
|
|
};
|
|
const result = await searchHfModels(searchParams);
|
|
console.debug("result", JSON.stringify(result));
|
|
setModelList(result);
|
|
};
|
|
|
|
return { modelList, getHuggingFaceModel };
|
|
}
|