PBThuong96 commited on
Commit
1b326bd
·
verified ·
1 Parent(s): 45b505c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
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-22.5-flash", temperature=0.2, google_api_key=GOOGLE_API_KEY)
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: