Update app.py

app.py

@@ -48,7 +48,7 @@ def get_predictions(text, mode):
  max_length=100,
  )
  elif mode == "Polish - RfC Detection":
- output = model.generate(
+ output = model_det_pl.generate(
  input_ids,
  no_repeat_ngram_size=2,
  num_beams=3,
@@ -91,15 +91,15 @@ if __name__ == "__main__":
 
  result = st.button("Find reason for contact")
  if mode == "Polish - RfC Generation (accepts whole conversation)":
- print("You selected RfC Generation model.")
- print("-- Input: Whole conversation. Should specify roles (e.g. AGENT: Hello, how can I help you? CLIENT: Hi, I would like to report a stolen card.")
- print("-- Output: Reason for calling for the whole conversation.")
+ st.markdown("### You selected RfC Generation model.")
+ st.markdown("-- *Input*: Whole conversation. Should specify roles (e.g. **AGENT: Hello, how can I help you? CLIENT: Hi, I would like to report a stolen card.**")
+ st.markdown("-- *Output*: Reason for calling for the whole conversation.")
  text_area = "Put a whole conversation or full e-mail here."
  
  elif mode == "Polish - RfC Detection (accepts one turn)":
- print("You selected RfC Detection model.")
- print("-- Input: A single turn from the conversation e.g. 'Hello, how can I help you?' or 'Hi, I would like to report a stolen card.'")
- print("-- Output: Model will return an empty string if a turn possibly does not includes Reason for Calling, or a sentence if the RfC is detected.")
+ st.markdown("### You selected RfC Detection model.")
+ st.markdown("-- *Input*: A single turn from the conversation e.g. **'Hello, how can I help you?'** or **'Hi, I would like to report a stolen card.'**")
+ st.markdown("-- *Output*: Model will return an empty string if a turn possibly does not includes Reason for Calling, or a sentence if the RfC is detected.")
  text_area = "Put a single turn or a few sentences here."
  
  if result: