Update app.py

app.py

@@ -15,7 +15,7 @@ model_name = "Llama-2-7b-MOM_Summar.Q2_K.gguf"
 # hf_hub_download(repo_id="sasvata/Llama2-7b-MOM-Summary-Finetuned-GGUF", filename=model_name, local_dir=model_path, local_dir_use_symlinks=False)
 
 print("Start the model init process")
-model = model = GPT4All(model_name, model_path, allow_download = False, device="cpu")
+model = GPT4All(model_name, model_path, allow_download = False, device="cpu")
 print("Finish the model init process")
 
 ## Function to convert plain text to markdown format
@@ -41,19 +41,13 @@ def generate_prompt(
 
 # Function to generate summary with the help of fine-tuned model
 def Summary_Gen(Transcript):
-    print("Summary Gen Fucntion Start: ", Transcript)
     prompt = generate_prompt(Transcript)
-    summary = model.generate(prompt=prompt,max_tokens=4096)
-    # summary_output = to_markdown(summary)
-    print("Summary Gen Fucntion End: ", summary)
+    summary = model.generate(prompt=prompt,max_tokens=1024)
     return summary
 
 # Function for text summarization
 def summarize_text(text):
-    print("Summarize Text Function Start: ", text)
-    # Your text summarization logic here (replaced with Summary_Gen)
     summarized_text = Summary_Gen(text)
-    print("Summarize Text Function End: ", summarized_text)
     return summarized_text
 
 
@@ -74,11 +68,10 @@ col1.title('Input')
 col2.title('Output')   
 
 col1.container(height=500, border=True).text_area("", option, height=1000)
+col2.container(height=500, border=True).markdown("", unsafe_allow_html=True)
 if col1.button("Summarize"):
     with st.spinner('Wait for it...'):
-        #print("Summary Function Start", summary_output)
         summary_output = summarize_text(option)
-        print("Summary Function End", summary_output)
         col2.container(height=500, border=True).markdown(summary_output, unsafe_allow_html=True)