prompt = "梅友乾明知其無資力支付酒店消費，亦無付款意願，竟意圖為自己不法之所有，"
query_dict = {
"inputs": prompt,
}
text_len = 300
t = trange(text_len, desc= '生成例稿', leave=True)
for i in t:
response = query(query_dict)
try:
response_text = response[0]['generated_text']
query_dict["inputs"] = response_text
t.set_description(f"{i}: {response[0]['generated_text']}")
t.refresh()
except KeyError:
sleep(30) # 如果伺服器太忙無回應，等30秒後再試。
pass
print(response[0]['generated_text'])

tokenizer = AutoTokenizer.from_pretrained("jslin09/gpt2-chinese-cluecorpussmall-finetuned-fraud")
model = AutoModelForCausalLM.from_pretrained("jslin09/gpt2-chinese-cluecorpussmall-finetuned-fraud")