Update Discussion_Buddy.py

Discussion_Buddy.py

@@ -105,7 +105,7 @@ def cleaning_text(text, is_lower_case = False):
 
 #Step 1 to classify, summary and get NER from news
 def extract_article(url):
-   model = genai.GenerativeModel('gemini-1.5-pro')
+   model = genai.GenerativeModel('gemini-1.5-flash')
    extract = model.generate_content(f"Extract the content from this news article: {url}, make sure extract all news part. Extract based on the news language used, if the news using Indonesia language just extract with Indonesian language, and so on if the news using english language. Make sure minimum is 200 words and maximum is 250 words").text
    return extract
 def process_news_pipeline(news_text, user_language):
@@ -153,7 +153,7 @@ def process_user_response(response_type, response_content, user_language):
  
 #Step 3, make chatbot for discussion about the news
 #load model and memory, to save context while discuss with chatbot
-model = GoogleGenerativeAI(model="gemini-1.5-pro", temperature=0.1)
+model = GoogleGenerativeAI(model="gemini-1.5-flash", temperature=0.1)
 memory = ConversationBufferMemory()
 conversation = ConversationChain(
     llm=model,