Views
No views yet
gemma-2b-quiz-kotransformers library:1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained("DORAEMONG/gemma-2b-quiz-ko")
5tokenizer = AutoTokenizer.from_pretrained("DORAEMONG/gemma-2b-quiz-ko")
6
7# Input a quiz question
8question = "다음 수학 문제의 답은 무엇입니까? 스피너가 A, B, C로 나뉘어 있을 때..."
9
10inputs = tokenizer(question, return_tensors="pt")
11outputs = model.generate(**inputs, max_new_tokens=100)
12
13# Decode the generated text
14print(tokenizer.decode(outputs[0], skip_special_tokens=True))