Update main.py

main.py

@@ -147,7 +147,7 @@ def parse_article_all(url: str) -> Dict[str, Any]:
 # ---------------------------------------
 def gemini_use(text_for_company: str) -> str:
     generation_config = genai.GenerationConfig(temperature=1)
-    model = genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
+    model = genai.GenerativeModel('gemini-1.5-flash-lite', generation_config=generation_config)
     prompt = f"""
     아래 내용을 참고해서 가장 연관성이 높은 주식 상장 회사 이름 하나만 말해줘.
     다른 설명 없이 회사 이름만 대답해.