Update app.py

app.py

@@ -145,7 +145,7 @@ def get_response_from_pdf(query, model, temperature=0.7):
 Write a detailed and complete response that answers the following user question: '{query}'
 Do not include a list of sources in your response. [/INST]"""
 
-    generated_text = generate_chunked_response(prompt, temperature=temperature)
+    generated_text = generate_chunked_response(prompt, model, temperature=temperature)
 
     # Clean the response
     clean_text = re.sub(r'<s>\[INST\].*?\[/INST\]\s*', '', generated_text, flags=re.DOTALL)
@@ -163,7 +163,7 @@ def get_response_with_search(query, model, temperature=0.7):
 Write a detailed and complete research document that fulfills the following user request: '{query}'
 After writing the document, please provide a list of sources used in your response. [/INST]"""
     
-    generated_text = generate_chunked_response(prompt, temperature=temperature)
+    generated_text = generate_chunked_response(prompt, model, temperature=temperature)
     
     # Clean the response
     clean_text = re.sub(r'<s>\[INST\].*?\[/INST\]\s*', '', generated_text, flags=re.DOTALL)
@@ -220,8 +220,8 @@ with gr.Blocks() as demo:
     )
 
     submit.click(chatbot_interface, 
-                 inputs=[msg, chatbot, use_web_search, model_dropdown, temperature_slider], 
-                 outputs=[chatbot])
+             inputs=[msg, chatbot, use_web_search, model_dropdown, temperature_slider], 
+             outputs=[chatbot])
     msg.submit(chatbot_interface, 
                inputs=[msg, chatbot, use_web_search, model_dropdown, temperature_slider], 
                outputs=[chatbot])