MohammedAlakhras commited on
Commit
d9c7450
·
verified ·
1 Parent(s): acd5c33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -6
app.py CHANGED
@@ -16,9 +16,6 @@ model = model.to(device)
16
 
17
  def answer_question(
18
  prompt,
19
- temperature=0.1,
20
- top_p=0.75,
21
- top_k=40,
22
  num_beams=2,
23
  **kwargs,
24
  ):
@@ -39,9 +36,6 @@ def answer_question(
39
  input_ids = inputs["input_ids"]
40
  attention_mask = inputs["attention_mask"]
41
  generation_config = GenerationConfig(
42
- temperature=temperature,
43
- top_p=top_p,
44
- top_k=top_k,
45
  num_beams=num_beams,
46
  **kwargs,
47
  )
 
16
 
17
  def answer_question(
18
  prompt,
 
 
 
19
  num_beams=2,
20
  **kwargs,
21
  ):
 
36
  input_ids = inputs["input_ids"]
37
  attention_mask = inputs["attention_mask"]
38
  generation_config = GenerationConfig(
 
 
 
39
  num_beams=num_beams,
40
  **kwargs,
41
  )