Update app.py

app.py

@@ -373,6 +373,16 @@ def handle_text_models(models, model_key, model_name):
                 try:
                     if model_key == 'ner':
                         result = models[model_key](input_text)
+                    elif model_key== 'text_generation':
+                        result = models[model_key](
+                            input_text,
+                            max_new_tokens=advanced_params.get('max_length', 100),
+                            do_sample=True,
+                            temperature=advanced_params.get('temperature', 0.7),
+                            top_k=50,
+                            top_p=0.95,
+                            num_return_sequences=advanced_params.get('num_return_sequences', 1)
+                        )
                     else:
                         result = models[model_key](input_text, **advanced_params)