Leesn465 commited on
Commit
48a0134
·
verified ·
1 Parent(s): 052c553

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -151,7 +151,7 @@ for m in genai.list_models():
151
 
152
  def gemini_use(text_for_company: str) -> str:
153
  generation_config = genai.GenerationConfig(temperature=1)
154
- model = genai.GenerativeModel('gemini-1.5-flash-lite', generation_config=generation_config)
155
  prompt = f"""
156
  아래 내용을 참고해서 가장 연관성이 높은 주식 상장 회사 이름 하나만 말해줘.
157
  다른 설명 없이 회사 이름만 대답해.
 
151
 
152
  def gemini_use(text_for_company: str) -> str:
153
  generation_config = genai.GenerationConfig(temperature=1)
154
+ model = genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
155
  prompt = f"""
156
  아래 내용을 참고해서 가장 연관성이 높은 주식 상장 회사 이름 하나만 말해줘.
157
  다른 설명 없이 회사 이름만 대답해.