msy127 commited on
Commit
4339c07
·
1 Parent(s): d7cb645

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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