GImInI commited on
Commit
ce16e7c
1 Parent(s): 00e025f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -25,7 +25,10 @@ def get_pdf_text(pdf_docs):
25
  # 과제
26
  # 아래 텍스트 추출 함수를 작성
27
  def get_text_file(docs):
28
- return docs.read()
 
 
 
29
 
30
  def get_csv_file(docs):
31
  csv_loader = CSVLoader(docs)
 
25
  # 과제
26
  # 아래 텍스트 추출 함수를 작성
27
  def get_text_file(docs):
28
+ if hasattr(docs, 'read'):
29
+ return docs.read()
30
+ elif isinstance(docs, str): # 만약 이미 텍스트인 경우
31
+ return docs
32
 
33
  def get_csv_file(docs):
34
  csv_loader = CSVLoader(docs)