Fix gen_kw bug

app.py

@@ -29,7 +29,7 @@ class TextGeneration:
             self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
             self.generator = pipeline(self.task, model=self.model_name_or_path, tokenizer=self.model_name_or_path)
 
-    def generate(self, prompt, **generation_kwargs):
+    def generate(self, prompt, generation_kwargs):
         if not self.debug:
             generation_kwargs["num_return_sequences"] = 1
 
@@ -132,7 +132,7 @@ def main():
             generation_kwargs_ph.markdown(", ".join([f"`{k}`: {v}" for k, v in generation_kwargs.items()]))
             text = normalize(text)
             if text:
-                generated_text = generator.generate(text)
+                generated_text = generator.generate(text, generation_kwargs)
                 st.markdown(
                     f'<p class="rtl rtl-box">'
                     f'<span class="result-text">{text} <span>'