tkdehf2 commited on
Commit
0b573ca
ยท
verified ยท
1 Parent(s): aea78f3

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -24
app.py DELETED
@@ -1,24 +0,0 @@
1
- import streamlit as st
2
- from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
-
4
- def app():
5
- st.title("์ž๋™ ์ผ๊ธฐ ์ƒ์„ฑ๊ธฐ")
6
-
7
- keywords = st.text_input("5๊ฐœ์˜ ํ‚ค์›Œ๋“œ๋ฅผ ์ž…๋ ฅํ•˜์„ธ์š” (์‰ผํ‘œ๋กœ ๊ตฌ๋ถ„)", "")
8
- keyword_list = [kw.strip() for kw in keywords.split(",")]
9
-
10
- if len(keyword_list) == 5 and st.button("์ผ๊ธฐ ์“ฐ๊ธฐ"):
11
- # ๋ชจ๋ธ ๋ฐ ํ† ํฌ๋‚˜์ด์ € ๋กœ๋“œ
12
- model = GPT2LMHeadModel.from_pretrained("gpt2")
13
- tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
14
-
15
- # ํ‚ค์›Œ๋“œ ๊ธฐ๋ฐ˜ fine-tuning
16
- input_ids = tokenizer.encode(" ".join(keyword_list), return_tensors="pt")
17
- output = model.generate(input_ids, max_length=500, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=5)
18
-
19
- # ์ƒ์„ฑ๋œ ์ผ๊ธฐ ์ถœ๋ ฅ
20
- diary = tokenizer.decode(output[0], skip_special_tokens=True)
21
- st.write(diary)
22
-
23
- if __name__ == "__main__":
24
- app()