lingy commited on
Commit
52b1f13
·
1 Parent(s): c129704

remove all printing

Browse files
src/content/__pycache__/common.cpython-312.pyc CHANGED
Binary files a/src/content/__pycache__/common.cpython-312.pyc and b/src/content/__pycache__/common.cpython-312.pyc differ
 
src/content/agent.py CHANGED
@@ -99,12 +99,10 @@ def _prepare_final_prompt_with_ui(one_time_prompt):
99
  return LLM_NO_AUDIO_PROMPT_TEMPLATE.format(user_question=one_time_prompt)
100
 
101
  with st.spinner("Searching appropriate querys..."):
102
- print(one_time_prompt)
103
  response = requests.get(
104
  f"{API_BASE_URL}retrieve_relevant_docs",
105
  params={"user_question": one_time_prompt}
106
  )
107
- print(response)
108
  relevant_query_indices = response.json()
109
 
110
  if len(st.session_state.ag_messages) <= 2:
 
99
  return LLM_NO_AUDIO_PROMPT_TEMPLATE.format(user_question=one_time_prompt)
100
 
101
  with st.spinner("Searching appropriate querys..."):
 
102
  response = requests.get(
103
  f"{API_BASE_URL}retrieve_relevant_docs",
104
  params={"user_question": one_time_prompt}
105
  )
 
106
  relevant_query_indices = response.json()
107
 
108
  if len(st.session_state.ag_messages) <= 2:
src/content/common.py CHANGED
@@ -68,9 +68,14 @@ MODEL_NAMES = OrderedDict({
68
  "ui_name": "LLM"
69
  },
70
  "audiollm": {
 
 
 
 
 
71
  "vllm_name": "MERaLiON/MERaLiON-AudioLLM-Whisper-SEA-LION-it",
72
  "model_name": "MERaLiON-AudioLLM-Whisper-SEA-LION-it",
73
- "ui_name": "AudioLLM"
74
  }
75
  })
76
 
@@ -540,7 +545,7 @@ def retrive_response_with_ui(
540
  }
541
 
542
  # print(request_data)
543
- print(model_name)
544
 
545
  error_msg = ""
546
  warnings = []
 
68
  "ui_name": "LLM"
69
  },
70
  "audiollm": {
71
+ "vllm_name": "MERaLiON/MERaLiON-AudioLLM-Whisper-SEA-LION",
72
+ "model_name": "MERaLiON-AudioLLM-Whisper-SEA-LION",
73
+ "ui_name": "AudioLLM"
74
+ },
75
+ "audiollm-it": {
76
  "vllm_name": "MERaLiON/MERaLiON-AudioLLM-Whisper-SEA-LION-it",
77
  "model_name": "MERaLiON-AudioLLM-Whisper-SEA-LION-it",
78
+ "ui_name": "AudioLLM-Instruction-Tuning"
79
  }
80
  })
81
 
 
545
  }
546
 
547
  # print(request_data)
548
+ # print(model_name)
549
 
550
  error_msg = ""
551
  warnings = []