Spaces:
Paused
Paused
change to google 2b model
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def create_knowledge_base(chunks):
|
|
37 |
|
38 |
# Hugging Face 모델 로드
|
39 |
def load_model():
|
40 |
-
model_name = "
|
41 |
tokenizer = AutoTokenizer.from_pretrained(model_name, clean_up_tokenization_spaces=False)
|
42 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
43 |
return pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=150, temperature=0.1)
|
|
|
37 |
|
38 |
# Hugging Face 모델 로드
|
39 |
def load_model():
|
40 |
+
model_name = "google/gemma-2-2b" # 텍스트 생성 모델 사용
|
41 |
tokenizer = AutoTokenizer.from_pretrained(model_name, clean_up_tokenization_spaces=False)
|
42 |
model = AutoModelForCausalLM.from_pretrained(model_name)
|
43 |
return pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=150, temperature=0.1)
|