karthikeyan-r commited on
Commit
0464246
·
verified ·
1 Parent(s): edf670f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -13,7 +13,7 @@ clear_conversation_button = st.sidebar.button("Clear Conversation")
13
  clear_model_button = st.sidebar.button("Clear Model")
14
 
15
  # Main UI
16
- st.title("Custom Trained Model Chat Conversation UI")
17
 
18
  # Session states
19
  if "model" not in st.session_state:
@@ -54,12 +54,10 @@ if st.session_state["qa_pipeline"]:
54
  user_input = st.text_input("Enter your query:", key="chat_input")
55
  if st.button("Send"):
56
  if user_input:
57
- st.write(f"Debug: Query - {user_input}") # Debugging
58
  with st.spinner("Generating response..."):
59
  try:
60
  response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=300)
61
  generated_text = response[0]["generated_text"]
62
- st.write(f"Debug: Response - {generated_text}") # Debugging
63
  st.session_state["conversation"].append(("You", user_input))
64
  st.session_state["conversation"].append(("Model", generated_text))
65
  except Exception as e:
 
13
  clear_model_button = st.sidebar.button("Clear Model")
14
 
15
  # Main UI
16
+ st.title("Chat Conversation")
17
 
18
  # Session states
19
  if "model" not in st.session_state:
 
54
  user_input = st.text_input("Enter your query:", key="chat_input")
55
  if st.button("Send"):
56
  if user_input:
 
57
  with st.spinner("Generating response..."):
58
  try:
59
  response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=300)
60
  generated_text = response[0]["generated_text"]
 
61
  st.session_state["conversation"].append(("You", user_input))
62
  st.session_state["conversation"].append(("Model", generated_text))
63
  except Exception as e: