Set false the full text result

app.py

@@ -36,6 +36,8 @@ class TextGeneration:
             max_length = len(self.tokenizer(prompt)["input_ids"]) + generation_kwargs["max_length"]
             generation_kwargs["max_length"] = max_length
 
+            generation_kwargs["return_full_text"] = False
+
             return self.generator(
                 prompt,
                 **generation_kwargs,