from summary_reverse_pred_native import * import gradio as gr import os text0 = "飓风格特是1993年9月在墨西哥和整个中美洲引发严重洪灾的大规模热带气旋,源于9月14日西南加勒比海上空一股东风波。次日从尼加拉瓜登岸,经过洪都拉斯后于9月17日在洪都拉斯湾再次达到热带风暴标准,但次日进入伯利兹上空后就减弱成热带低气压。穿过尤卡坦半岛后,在9月20日强化成二级飓风,从韦拉克鲁斯州的图斯潘附近登陆墨西哥。9月21日从纳亚里特州进入太平洋时已降级成热带低气压,最终于5天后在开放水域上空消散。" text1 = "珊瑚坝是长江中的一处河漫滩,位于长江重庆市渝中区区段主航道左侧[1],靠近渝中半岛,原分属重庆市市中区菜园坝街道和石板坡街道[2],现属渝中区菜园坝街道石板坡社区[3],是长江上游缓冲地段自然冲积沙洲,略呈纺锤形[4]或椭圆形,长约1800米,宽约600米,坝上遍布鹅卵石和水草。每年夏季洪水时均被淹没,其余时间常露水面,枯水期则与长江左岸相连[5]。" example_sample = [ [text0, False], [text1, False], ] def demo_func(prefix, do_sample): l = simple_pred(prefix, do_sample = do_sample) return { "Dialogue Context": l } demo = gr.Interface( fn=demo_func, inputs=[gr.Text(label = "Context"), gr.Checkbox(label="do sample"), ], outputs="json", title=f"Chinese Context Dialogue Generator 🐰 demonstration", examples=example_sample if example_sample else None, cache_examples = False ) demo.launch(server_name=None, server_port=None)