Hanjaewon commited on
Commit
9b93859
ยท
verified ยท
1 Parent(s): c460da1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +31 -0
app.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ pip install transformers gradio
2
+ import gradio as gr
3
+ from transformers import T5ForConditionalGeneration, T5Tokenizer
4
+
5
+ # Hugging Face์˜ t5-base ๋ชจ๋ธ ๋ฐ ํ† ํฌ๋‚˜์ด์ €๋ฅผ ๋ถˆ๋Ÿฌ์˜ต๋‹ˆ๋‹ค.
6
+ model_name = "t5-base"
7
+ model = T5ForConditionalGeneration.from_pretrained(model_name)
8
+ tokenizer = T5Tokenizer.from_pretrained(model_name)
9
+
10
+ def generate_question(prompt):
11
+ # ์ฃผ์–ด์ง„ ์ž…๋ ฅ์— ๊ธฐ๋ฐ˜ํ•˜์—ฌ ํ€ด์ฆˆ๋ฅผ ์ƒ์„ฑํ•˜๋Š” ํ•จ์ˆ˜
12
+ input_text = f"ํ€ด์ฆˆ: {prompt} ๋Œ€๋‹ต:"
13
+ input_ids = tokenizer.encode(input_text, return_tensors="pt")
14
+
15
+ # ๋ชจ๋ธ์„ ์‚ฌ์šฉํ•˜์—ฌ ํ€ด์ฆˆ ์ƒ์„ฑ
16
+ output = model.generate(input_ids)
17
+ question = tokenizer.decode(output[0], skip_special_tokens=True)
18
+
19
+ return question
20
+
21
+ # Gradio๋ฅผ ์‚ฌ์šฉํ•˜์—ฌ ์›น ์ธํ„ฐํŽ˜์ด์Šค๋ฅผ ๋งŒ๋“ญ๋‹ˆ๋‹ค.
22
+ iface = gr.Interface(
23
+ fn=generate_question,
24
+ inputs="text",
25
+ outputs="text",
26
+ live=True,
27
+ interpretation="default"
28
+ )
29
+
30
+ # ์›น ์„œ๋น„์Šค๋ฅผ ์‹œ์ž‘ํ•ฉ๋‹ˆ๋‹ค.
31
+ iface.launch()