Update app.py

app.py

@@ -16,7 +16,8 @@ def generate(starting_text, choice, temp, top_p):
     elif choice == '25M':
         model = tiny
     elif choice == "Info":
-        return desc
+        yield desc
+        return
     
     results = []
     for i in range(5):
@@ -31,7 +32,7 @@ def generate(starting_text, choice, temp, top_p):
         outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
         outputs = outputs[:outputs.find(".")]
         results.append(outputs)
-    return "\n\n".join(results)
+        yield "\n\n".join(results)
 
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     starting_text = gr.Textbox(label="Starting text", value="cinta adalah")