Update app.py

app.py

@@ -41,10 +41,10 @@ def summarize(file, n_words):
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=3000, chunk_overlap=200)
     chunks = text_splitter.create_documents([text])
     n_words = n_words
-    template = ''' [INST]
+    template = '''
     Your task is to summarize a long text into a concise summary of a specific number of words. 
 
-    The summary you generate must be EXACTLY {N_WORDS} words long. 
+    The summary you generate must be very detailed. 
 
     Before writing your final summary, first break down the key points of the text in a <scratchpad>. Identify the most important information that should be included in a summary of the specified length.
 
@@ -53,13 +53,10 @@ def summarize(file, n_words):
     Here is the long text to summarize:
     Text: 
     {TEXT}
-
-
-    [/INST]
     '''
     prompt = PromptTemplate(
         template=template,
-        input_variables=['TEXT', "N_WORDS"]    
+        input_variables=['TEXT']    
     )
     formatted_prompt = prompt.format(TEXT=text, N_WORDS=n_words)
     output_summary = llm_engine_hf.invoke(formatted_prompt)
@@ -83,14 +80,13 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            n_words = gr.Slider(minimum=50, maximum=500, step=50, label="Number of words (approximately)")
             file = gr.File(label="Submit a file")
         
         with gr.Column():
             output_text = gr.Textbox(label="Summary", lines=20)
 
     submit_button = gr.Button("Summarize")
-    submit_button.click(summarize, inputs=[file, n_words], outputs=output_text)
+    submit_button.click(summarize, inputs=[file], outputs=output_text)
 
     def generate_file():
         summary_text = output_text