SergeyO7 commited on
Commit
309813d
·
verified ·
1 Parent(s): 351956c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -91,11 +91,9 @@ def process_query(query_text: str, vectorstore):
91
  model = HuggingFaceEndpoint(
92
  repo_id="cointegrated/rut5-base",
93
  task="text2text-generation",
94
- model_kwargs={
95
- "temperature": 0.5,
96
- "max_length": 512
97
- }
98
- # # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN") # Раскомментируйте, если нужен токен
99
  )
100
 
101
  # Инициализация модели с базовыми параметрами
 
91
  model = HuggingFaceEndpoint(
92
  repo_id="cointegrated/rut5-base",
93
  task="text2text-generation",
94
+ temperature=0.5,
95
+ max_length=512,
96
+ repetition_penalty=1.2
 
 
97
  )
98
 
99
  # Инициализация модели с базовыми параметрами