tkdehf2 commited on
Commit
77a648d
ยท
verified ยท
1 Parent(s): a8eea57

Delete app.py

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