Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,9 @@ from typing import Any, Optional, Dict, List
|
|
23 |
from huggingface_hub import InferenceClient
|
24 |
from langchain.llms.base import LLM
|
25 |
|
|
|
|
|
|
|
26 |
class CustomInferenceClient(LLM, KwArgsModel):
|
27 |
model_name: str
|
28 |
inference_client: InferenceClient
|
|
|
23 |
from huggingface_hub import InferenceClient
|
24 |
from langchain.llms.base import LLM
|
25 |
|
26 |
+
class KwArgsModel(BaseModel):
|
27 |
+
kwargs: Dict[str, Any] = Field(default_factory=dict)
|
28 |
+
|
29 |
class CustomInferenceClient(LLM, KwArgsModel):
|
30 |
model_name: str
|
31 |
inference_client: InferenceClient
|