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)
|