sys2 commited on
Commit
2183dc5
1 Parent(s): 632eab2

first commit

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -13,8 +13,6 @@ from langchain.llms import HuggingFaceHub, LlamaCpp, CTransformers # For loadin
13
  from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
14
  import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
15
  import os
16
- import pandas as pd
17
- import json
18
 
19
 
20
  # PDF 문서로부터 텍스트를 추출하는 함수입니다.
@@ -49,9 +47,9 @@ def get_csv_file(csv_docs):
49
  with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
50
  f.write(csv_docs.getvalue())
51
 
52
- text_loader = CSVLoader(temp_filepath)
53
- text_doc = text_loader.load()
54
- return text_doc
55
 
56
  def get_json_file(json_docs):
57
  temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
 
13
  from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
14
  import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
15
  import os
 
 
16
 
17
 
18
  # PDF 문서로부터 텍스트를 추출하는 함수입니다.
 
47
  with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
48
  f.write(csv_docs.getvalue())
49
 
50
+ csv_loader = CSVLoader(temp_filepath)
51
+ csv_doc = csv_loader.load()
52
+ return csv_doc
53
 
54
  def get_json_file(json_docs):
55
  temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.