tkdehf2 commited on
Commit
3347488
โ€ข
1 Parent(s): 18ecda7

Update app.py

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