SimrusDenuvo commited on
Commit
cdbe614
·
verified ·
1 Parent(s): 13917a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def generate_response(prompt):
26
  eos_token_id=tokenizer.eos_token_id
27
  )
28
 
29
- response = tokenizer.decode(output[0], skip_special_tokens=True)
30
 
31
  # Убираем часть промпта из начала
32
  return response.replace(instruction, "").strip()
 
26
  eos_token_id=tokenizer.eos_token_id
27
  )
28
 
29
+ response = generator(prompt, max_length=100, num_return_sequences=1, temperature=0.7, top_k=50, top_p=0.95)
30
 
31
  # Убираем часть промпта из начала
32
  return response.replace(instruction, "").strip()