yirmibesogluz commited on
Commit
c93dad0
1 Parent(s): 6805c60

Fixed default argument error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="turna_summarization_tr_news", 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)
 
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)