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

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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()