Geraldine commited on
Commit
04f4640
·
verified ·
1 Parent(s): 95067c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -319,8 +319,7 @@ with main_col:
319
 
320
  # Process for Tab 4 - Fine-tuned model
321
  with tab4:
322
- llm = create_aws_ollama_llm()
323
- fine_tuned_chain = setup_fine_tuned_tab(llm)
324
  st.session_state.response_fine_tuned = fine_tuned_chain.invoke(query)
325
  with st.chat_message("assistant"):
326
  st.markdown(st.session_state.response_fine_tuned)
 
319
 
320
  # Process for Tab 4 - Fine-tuned model
321
  with tab4:
322
+ fine_tuned_chain = setup_fine_tuned_tab()
 
323
  st.session_state.response_fine_tuned = fine_tuned_chain.invoke(query)
324
  with st.chat_message("assistant"):
325
  st.markdown(st.session_state.response_fine_tuned)