Update main.py

main.py

@@ -145,6 +145,10 @@ def parse_article_all(url: str) -> Dict[str, Any]:
 # ---------------------------------------
 # 회사명 추론 (Gemini)
 # ---------------------------------------
+for m in genai.list_models():
+    if 'generateContent' in m.supported_generation_methods:
+        print(m.name)
+
 def gemini_use(text_for_company: str) -> str:
     generation_config = genai.GenerationConfig(temperature=1)
     model = genai.GenerativeModel('gemini-1.5-flash-lite', generation_config=generation_config)