Chan-Y commited on
Commit
1dcae0c
1 Parent(s): 241247a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -5,7 +5,6 @@ import PyPDF2
5
  import gradio as gr
6
  from langchain.prompts import PromptTemplate
7
  from langchain.chains.summarize import load_summarize_chain
8
- from langchain_core.documents import Document
9
  from pathlib import Path
10
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
11
 
@@ -61,12 +60,10 @@ Using the essay summary and the answers to the interesting questions, create a c
61
  input_variables=['TEXT']
62
  )
63
 
64
- summary = ""
65
- while len(summary.split()) < 100:
66
- formatted_prompt = prompt.format(TEXT=text)
67
- output_summary = llm_engine_hf.invoke(formatted_prompt)
68
- summary = output_summary.content
69
- return summary
70
 
71
  def download_summary(output_text):
72
  if output_text:
 
5
  import gradio as gr
6
  from langchain.prompts import PromptTemplate
7
  from langchain.chains.summarize import load_summarize_chain
 
8
  from pathlib import Path
9
  from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
10
 
 
60
  input_variables=['TEXT']
61
  )
62
 
63
+ formatted_prompt = prompt.format(TEXT=text)
64
+ output_summary = llm_engine_hf.invoke(formatted_prompt)
65
+
66
+ return output_summary.content
 
 
67
 
68
  def download_summary(output_text):
69
  if output_text: