rairo commited on
Commit
3a7575a
·
verified ·
1 Parent(s): ed4e2b3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -283,7 +283,8 @@ def generate_story_endpoint():
283
 
284
  # Convert PDF to text
285
  pdf_text = get_pdf_text(uploaded_file)
286
- full_story = generate_story_from_text(pdf_text, story_type)
 
287
 
288
  elif input_type == "dataframe":
289
  # Expecting a file in request.files["file"] and an "ext" field (csv, xlsx, xls)
 
283
 
284
  # Convert PDF to text
285
  pdf_text = get_pdf_text(uploaded_file)
286
+ prompt = pdf_text
287
+ full_story = generate_story_from_text(prompt, story_type)
288
 
289
  elif input_type == "dataframe":
290
  # Expecting a file in request.files["file"] and an "ext" field (csv, xlsx, xls)