sahilursa commited on
Commit
f0cb788
Β·
verified Β·
1 Parent(s): ec52082

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ if os.path.exists(vector_store_file) and os.path.exists(data_file):
46
  # Load the embedding and LLM models
47
  print("Loading AI and embedding models...")
48
  embedding_model = SentenceTransformer('BAAI/bge-large-en-v1.5')
49
- llm = genai.GenerativeModel('gemini-pro') # Using gemini-pro as it's a robust choice
50
  print("βœ… Models loaded successfully.")
51
 
52
  except Exception as e:
 
46
  # Load the embedding and LLM models
47
  print("Loading AI and embedding models...")
48
  embedding_model = SentenceTransformer('BAAI/bge-large-en-v1.5')
49
+ llm = genai.GenerativeModel('gemini-2.5-flash') # Using gemini-pro as it's a robust choice
50
  print("βœ… Models loaded successfully.")
51
 
52
  except Exception as e: