Spaces:
Sleeping
Sleeping
Commit
·
fafce48
1
Parent(s):
ab70839
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from langchain.prompts import PromptTemplate
|
|
| 11 |
|
| 12 |
DEVICE = 'cpu'
|
| 13 |
FILE_EXT = ['pdf','text','csv','word','wav']
|
| 14 |
-
DEFAULT_SYSTEM_PROMPT = "As a chatbot you are answering
|
| 15 |
MAX_NEW_TOKENS = 4096
|
| 16 |
DEFAULT_TEMPERATURE = 0.1
|
| 17 |
DEFAULT_MAX_NEW_TOKENS = 2048
|
|
@@ -49,7 +49,7 @@ def chat_application(llm_service,key,temperature=0.1,max_tokens=1024):
|
|
| 49 |
return llm
|
| 50 |
|
| 51 |
|
| 52 |
-
def document_loader(file_path,api_key,doc_type='pdf',llm='
|
| 53 |
document = None
|
| 54 |
if doc_type == 'pdf':
|
| 55 |
document = process_pdf_document(document_file=file_path)
|
|
|
|
| 11 |
|
| 12 |
DEVICE = 'cpu'
|
| 13 |
FILE_EXT = ['pdf','text','csv','word','wav']
|
| 14 |
+
DEFAULT_SYSTEM_PROMPT = "As a chatbot you are answering questions being requested."
|
| 15 |
MAX_NEW_TOKENS = 4096
|
| 16 |
DEFAULT_TEMPERATURE = 0.1
|
| 17 |
DEFAULT_MAX_NEW_TOKENS = 2048
|
|
|
|
| 49 |
return llm
|
| 50 |
|
| 51 |
|
| 52 |
+
def document_loader(file_path,api_key,doc_type='pdf',llm='HuggingFace',temperature=0.1,max_tokens=4096):
|
| 53 |
document = None
|
| 54 |
if doc_type == 'pdf':
|
| 55 |
document = process_pdf_document(document_file=file_path)
|