Spaces:
Paused
Paused
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 |
)
|