Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,7 +19,7 @@ transformers.logging.set_verbosity_error()
 st.set_page_config(page_title="Trình sinh tiêu đề", layout="centered")
 
 # Cấu hình Gemini API (thay YOUR_GEMINI_API_KEY bằng API key thực tế)
-GEMINI_API_KEY = "AIzaSyBE3FCLsq2HiaXIFtAckQODch5DstsTLBA"  # Thay bằng API key thực tế của bạn
+GEMINI_API_KEY = "AIzaSyCEDRquPDC9N09hTHGD9FfvsPP83AZT78Q"  # Thay bằng API key thực tế của bạn
 genai.configure(api_key=GEMINI_API_KEY)
 
 # Các mô hình
@@ -43,7 +43,7 @@ TITLE_MODELS = {
         "model_type": "seq2seq"
     },
     "Gemini Title Generator": {
-        "model_path": "gemini-1.5-pro",
+        "model_path": "gemini-1.5-flash",
         "tokenizer_path": None,
         "token": False,
         "model_type": "gemini"
@@ -64,7 +64,7 @@ SUMMARIZATION_MODELS = {
         "model_type": "seq2seq"
     },
     "Gemini Summarization": {
-        "model_path": "gemini-1.5-pro",
+        "model_path": "gemini-1.5-flash",
         "tokenizer_path": None,
         "token": False,
         "model_type": "gemini"