alaselababatunde commited on
Commit
3f1fdb5
·
1 Parent(s): c7c14f1
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -28,7 +28,7 @@ spitch_client = Spitch()
28
  llm = HuggingFaceEndpoint(
29
  repo_id=HF_MODEL,
30
  temperature=0.2,
31
- max_length=512
32
  )
33
 
34
  # FastAPI app
 
28
  llm = HuggingFaceEndpoint(
29
  repo_id=HF_MODEL,
30
  temperature=0.2,
31
+ model_kwargs={"max_length": 512}
32
  )
33
 
34
  # FastAPI app