Hugging Face's logo

Spaces:
prithivida
/
Gramformer
Running

App Files Community
1
prithivida commited on
Commit
2fa02ef
·
1 Parent(s): 988a245

Update InferenceServer.py

Browse files
Files changed (1) hide show
  1. InferenceServer.py +3 -3
InferenceServer.py CHANGED
@@ -47,9 +47,9 @@ def correct(input_sentence, max_candidates=1):
47
  input_ids,
48
  do_sample=True,
49
  max_length=128,
50
- # top_k=50,
51
- # top_p=0.95,
52
- num_beams=7,
53
  early_stopping=True,
54
  num_return_sequences=max_candidates)
55
 
 
47
  input_ids,
48
  do_sample=True,
49
  max_length=128,
50
+ top_k=50,
51
+ top_p=0.95,
52
+ # num_beams=7,
53
  early_stopping=True,
54
  num_return_sequences=max_candidates)
55