Brasd99 commited on
Commit
ee7c758
2 Parent(s): bf6aa9d 1e318ff

Merge branch 'main' of https://huggingface.co/spaces/Brasd99/AnswerMate into main

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def generate(
28
  model, tok, text,
29
  do_sample=True, max_length=10000, repetition_penalty=5.0,
30
  top_k=5, top_p=0.95, temperature=1,
31
- num_beams=None,
32
  no_repeat_ngram_size=3
33
  ):
34
  input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)
 
28
  model, tok, text,
29
  do_sample=True, max_length=10000, repetition_penalty=5.0,
30
  top_k=5, top_p=0.95, temperature=1,
31
+ num_beams=10,
32
  no_repeat_ngram_size=3
33
  ):
34
  input_ids = tok.encode(text, return_tensors="pt").to(DEVICE)