Spaces:
Running
on
Zero
Running
on
Zero
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)
|