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μ›” μ…‹μ§Έ μ£Ό λͺ©μš”일에 μ‹œν–‰λ©λ‹ˆλ‹€.