Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,16 +1,13 @@
|
|
| 1 |
import os
|
| 2 |
import sys
|
| 3 |
-
import logging
|
| 4 |
-
import chromadb
|
| 5 |
-
import gradio as gr
|
| 6 |
-
|
| 7 |
-
# Fix lỗi SQLite cũ trên Hugging Face
|
| 8 |
try:
|
| 9 |
__import__("pysqlite3")
|
| 10 |
sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
|
| 11 |
except ImportError:
|
| 12 |
pass
|
| 13 |
-
|
|
|
|
|
|
|
| 14 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 15 |
from langchain_chroma import Chroma
|
| 16 |
from langchain_huggingface import HuggingFaceEmbeddings
|
|
@@ -73,7 +70,7 @@ class DeepMedBot:
|
|
| 73 |
self.ready = False
|
| 74 |
try:
|
| 75 |
self.fast_retriever, self.deep_retriever = get_retrievers()
|
| 76 |
-
self.llm = ChatGoogleGenerativeAI(model="gemini-
|
| 77 |
self._build_chains()
|
| 78 |
self.ready = True
|
| 79 |
except Exception as e:
|
|
|
|
| 1 |
import os
|
| 2 |
import sys
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
try:
|
| 4 |
__import__("pysqlite3")
|
| 5 |
sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
|
| 6 |
except ImportError:
|
| 7 |
pass
|
| 8 |
+
import logging
|
| 9 |
+
import chromadb
|
| 10 |
+
import gradio as gr
|
| 11 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 12 |
from langchain_chroma import Chroma
|
| 13 |
from langchain_huggingface import HuggingFaceEmbeddings
|
|
|
|
| 70 |
self.ready = False
|
| 71 |
try:
|
| 72 |
self.fast_retriever, self.deep_retriever = get_retrievers()
|
| 73 |
+
self.llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0.2, google_api_key=GOOGLE_API_KEY)
|
| 74 |
self._build_chains()
|
| 75 |
self.ready = True
|
| 76 |
except Exception as e:
|