Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='wannaphong/mark2_d24')
 
 def generate(text):
-    return generator(text,max_new_tokens=100) # use yield from#result = generator(text, max_length=30, num_return_sequences=1)
+    return generator(text,max_new_tokens=100)[0]["generated_text"] # use yield from#result = generator(text, max_length=30, num_return_sequences=1)
     # return result[0]["generated_text"]  # type: ignore
 
 examples = [