Update app.py

app.py

@@ -99,11 +99,11 @@ def load_model(_docs):
                              model=model,
                              tokenizer=tokenizer,
                              max_new_tokens=1024,
-                             temperature=0.5,
+                             temperature=0.1,
                              top_p=0.95,
                              repetition_penalty=1.15,
                              streamer=streamer,)
-    llm = HuggingFacePipeline(pipeline=text_pipeline, model_kwargs={"temperature": 0.5})
+    llm = HuggingFacePipeline(pipeline=text_pipeline, model_kwargs={"temperature": 0.1})
 
     # SYSTEM_PROMPT = ("Use the following pieces of context to answer the question at the end. "
     #                  "If you don't know the answer, just say that you don't know, "
@@ -238,6 +238,7 @@ if prompt := st.chat_input("How can I help you today?"):
             generate_pdf()
             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
             st.image(st.session_state['reference'])
+            st.write('Book name')
 
         # if choice == 'TTS':
         #     with open('/home/user/app/audio/audio.mp3','wb') as sound_file: