aiTOYAMA commited on
Commit
c441e94
·
verified ·
1 Parent(s): f822eb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -1,16 +1,13 @@
1
  import os
2
  import shutil
3
  from datetime import datetime
4
-
5
  import gradio as gr
6
  from langchain_chroma import Chroma
7
  from langchain_community.document_loaders import PyPDFLoader
8
  from langchain_text_splitters import RecursiveCharacterTextSplitter
9
-
10
- from langchain_google_genai import (
11
- GoogleGenerativeAIEmbeddings,
12
- ChatGoogleGenerativeAI
13
- )
14
  from huggingface_hub import snapshot_download
15
 
16
  # ==================================================
@@ -35,8 +32,8 @@ if API_KEY is None:
35
  "GEMINI_API_KEY が設定されていません"
36
  )
37
 
38
- embedding_model = GoogleGenerativeAIEmbeddings(
39
- model="models/embedding-001"
40
  )
41
 
42
  llm = ChatGoogleGenerativeAI(
 
1
  import os
2
  import shutil
3
  from datetime import datetime
4
+ from google import genai
5
  import gradio as gr
6
  from langchain_chroma import Chroma
7
  from langchain_community.document_loaders import PyPDFLoader
8
  from langchain_text_splitters import RecursiveCharacterTextSplitter
9
+ from langchain_huggingface import HuggingFaceEmbeddings
10
+ from langchain_google_genai import ChatGoogleGenerativeAI
 
 
 
11
  from huggingface_hub import snapshot_download
12
 
13
  # ==================================================
 
32
  "GEMINI_API_KEY が設定されていません"
33
  )
34
 
35
+ embedding_model = HuggingFaceEmbeddings(
36
+ model="sentence-transformers/all-MiniLM-L6-v2"
37
  )
38
 
39
  llm = ChatGoogleGenerativeAI(