Update app.py

app.py

@@ -48,7 +48,8 @@ def infer(prompt, max_tokens, temperature, top_k, top_p):
         top_k=top_k,
         top_p=top_p,
         do_sample=True,
-        return_full_text=False
+        return_full_text=True,
+        skip_special_tokens=False
     )
     return result[0]["generated_text"] if result and "generated_text" in result[0] else ""
 
@@ -61,12 +62,12 @@ with gr.Blocks() as demo:
         temperature = gr.Slider(0.1, 2.0, value=1.0, step=0.01, label="溫度 (temperature)")
         top_k = gr.Slider(1, 100, value=40, step=1, label="Top-k")
         top_p = gr.Slider(0.0, 1.0, value=1.0, step=0.01, label="Top-p")
-    output = gr.Textbox(label="模型輸出", lines=20, max_lines=40, show_copy_button=True)
+    #output = gr.Textbox(label="模型輸出", lines=20, max_lines=40, show_copy_button=True)
     btn = gr.Button("送出")
     btn.click(
         infer,
         inputs=[prompt, max_tokens, temperature, top_k, top_p],
-        outputs=output
+        outputs=prompt
     )
 
 if __name__ == "__main__":