chef choice

generator.py

@@ -90,9 +90,10 @@ def skip_special_tokens_and_prettify(text):
     return data
 
 
-def generation_function(texts, lang="en"):
+def generation_function(texts, lang="en", chef="top"):
 
     langs = ['ru', 'en']
+    chefs = {'top': chef_top, 'beam': chef_beam}
     api_credentials = load_api()
 
     if lang != "en" and lang in langs:
@@ -100,7 +101,7 @@ def generation_function(texts, lang="en"):
 
     output_ids = generator(
         texts,
-        ** chef_top
+        ** chefs[chef]
     )[0]["generated_token_ids"]
 
     recepie = tokenizer.decode(output_ids, skip_special_tokens=False)