Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
|
4 |
# model = AutoModelForCausalLM.from_pretrained("Karzan/ckb-gpt2-medium-base-test-1024")
|
5 |
model_id = "Karzan/bart-qa-ckb"
|
6 |
-
pipe = pipeline("
|
7 |
|
8 |
|
9 |
def func(context, question):
|
|
|
3 |
|
4 |
# model = AutoModelForCausalLM.from_pretrained("Karzan/ckb-gpt2-medium-base-test-1024")
|
5 |
model_id = "Karzan/bart-qa-ckb"
|
6 |
+
pipe = pipeline("question-answering", model=model_id,max_answer_len=300,handle_impossible_answer=True,top_k=1)
|
7 |
|
8 |
|
9 |
def func(context, question):
|