widefirst103 commited on
Commit
660ede1
ยท
verified ยท
1 Parent(s): f0aa58c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ model_name = "nlpai-lab/kobert-base-v1"
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
7
 
8
- qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
9
 
10
  context = """
11
  ์ˆ˜๋Šฅ์€ ๋งค๋…„ 11์›” ์…‹์งธ ์ฃผ ๋ชฉ์š”์ผ์— ์‹œํ–‰๋ฉ๋‹ˆ๋‹ค.
 
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
7
 
8
+ generator = pipeline("text-generation", model="EleutherAI/polyglot-ko-1.3b", device_map="auto")
9
+
10
 
11
  context = """
12
  ์ˆ˜๋Šฅ์€ ๋งค๋…„ 11์›” ์…‹์งธ ์ฃผ ๋ชฉ์š”์ผ์— ์‹œํ–‰๋ฉ๋‹ˆ๋‹ค.