File size: 1,112 Bytes
5b8fd7e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import React, { createContext, useContext, useState } from 'react'
interface TextClassificationConfig {
top_k: number
}
interface TextClassificationContextType {
config: TextClassificationConfig
setConfig: React.Dispatch<React.SetStateAction<TextClassificationConfig>>
}
const TextClassificationContext = createContext<
TextClassificationContextType | undefined
>(undefined)
export function useTextClassification() {
const context = useContext(TextClassificationContext)
if (context === undefined) {
throw new Error(
'useTextClassification must be used within a TextClassificationProvider'
)
}
return context
}
interface TextClassificationProviderProps {
children: React.ReactNode
}
export function TextClassificationProvider({
children
}: TextClassificationProviderProps) {
const [config, setConfig] = useState<TextClassificationConfig>({
top_k: 1
})
const value: TextClassificationContextType = {
config,
setConfig
}
return (
<TextClassificationContext.Provider value={value}>
{children}
</TextClassificationContext.Provider>
)
}
|