NitinBot001 commited on
Commit
7600fa1
·
verified ·
1 Parent(s): 6d7bcf5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,9 +14,9 @@ from fastapi.responses import HTMLResponse, JSONResponse
14
  from pydantic import BaseModel, Field
15
  import uvicorn
16
 
17
- from langchain.document_loaders import TextLoader
18
  from langchain.text_splitter import RecursiveCharacterTextSplitter
19
- from langchain.vectorstores import FAISS
20
  from langchain.chains import RetrievalQA
21
  from langchain.prompts import PromptTemplate
22
  from langchain.callbacks.base import BaseCallbackHandler
@@ -59,7 +59,7 @@ class Config:
59
  CHUNK_OVERLAP = 100
60
  MAX_RETRIES = 3
61
  RATE_LIMIT_DELAY = 1.0
62
- MODEL_NAME = "gemini-1.5-flash"
63
  EMBEDDING_MODEL = "models/embedding-001"
64
  TEMPERATURE = 0.5
65
  MAX_OUTPUT_TOKENS = 512
 
14
  from pydantic import BaseModel, Field
15
  import uvicorn
16
 
17
+ from langchain_community.document_loaders import TextLoader
18
  from langchain.text_splitter import RecursiveCharacterTextSplitter
19
+ from langchain_community.vectorstores import FAISS
20
  from langchain.chains import RetrievalQA
21
  from langchain.prompts import PromptTemplate
22
  from langchain.callbacks.base import BaseCallbackHandler
 
59
  CHUNK_OVERLAP = 100
60
  MAX_RETRIES = 3
61
  RATE_LIMIT_DELAY = 1.0
62
+ MODEL_NAME = "gemma-3-27b-it"
63
  EMBEDDING_MODEL = "models/embedding-001"
64
  TEMPERATURE = 0.5
65
  MAX_OUTPUT_TOKENS = 512