Spaces:
Runtime error
Runtime error
MohammedAlakhras
commited on
Update app.py
Browse files
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 |
)
|