Update app.py

app.py

@@ -4,8 +4,8 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='wannaphong/mark2_d24')
 
 def generate(text):
-    result = generator(text, max_length=30, num_return_sequences=1)
-    return result[0]["generated_text"]  # type: ignore
+    yield from pipe(prompt,stream=True,max_new_tokens=50) # use yield from#result = generator(text, max_length=30, num_return_sequences=1)
+    # return result[0]["generated_text"]  # type: ignore
 
 examples = [
     ["The Moon's orbit around Earth has"],
@@ -19,4 +19,4 @@ demo = gr.Interface(
     examples=examples
 )
 
-demo.launch()
+demo.launch()