Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,8 +29,7 @@ retriever = vectordb.as_retriever(search_kwargs={"k": 5})
|
|
29 |
class KwArgsModel(BaseModel):
|
30 |
kwargs: Dict[str, Any] = Field(default_factory=dict)
|
31 |
|
32 |
-
class CustomInferenceClient(LLM, KwArgsModel):
|
33 |
-
__metaclass__ = type(LLM)
|
34 |
model_name: str
|
35 |
inference_client: InferenceClient
|
36 |
|
|
|
29 |
class KwArgsModel(BaseModel):
|
30 |
kwargs: Dict[str, Any] = Field(default_factory=dict)
|
31 |
|
32 |
+
class CustomInferenceClient(LLM, KwArgsModel, metaclass=type(LLM)):
|
|
|
33 |
model_name: str
|
34 |
inference_client: InferenceClient
|
35 |
|