Spaces:
Paused
Paused
Commit
·
c93dad0
1
Parent(s):
6805c60
Fixed default argument error
Browse files
app.py
CHANGED
|
@@ -110,7 +110,7 @@ def paraphrase(input, model_choice="turna_paraphrasing_tatoeba"):
|
|
| 110 |
return paraphrasing_sub(input)[0]["generated_text"]
|
| 111 |
|
| 112 |
@spaces.GPU
|
| 113 |
-
def summarize(input, model_choice
|
| 114 |
model_mapping = {"turna_summarization_tr_news": "boun-tabi-LMG/turna_summarization_tr_news",
|
| 115 |
"turna_summarization_mlsum": "boun-tabi-LMG/turna_summarization_mlsum"}
|
| 116 |
summarization_model = pipeline(model=model_mapping[model_choice], device=0)
|
|
|
|
| 110 |
return paraphrasing_sub(input)[0]["generated_text"]
|
| 111 |
|
| 112 |
@spaces.GPU
|
| 113 |
+
def summarize(input, model_choice, max_new_tokens, length_penalty, no_repeat_ngram_size):
|
| 114 |
model_mapping = {"turna_summarization_tr_news": "boun-tabi-LMG/turna_summarization_tr_news",
|
| 115 |
"turna_summarization_mlsum": "boun-tabi-LMG/turna_summarization_mlsum"}
|
| 116 |
summarization_model = pipeline(model=model_mapping[model_choice], device=0)
|