OrangeEye commited on
Commit
5361e2c
·
1 Parent(s): ac7f2bf

update llama

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -141,6 +141,7 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
141
  input = gr.Textbox(visible=False) # placeholder
142
  gr_md = gr.Markdown(mark_text + md_text_initial)
143
 
 
144
  def update_with_rag_md(message, llm_results_use = 5, database_choice = index_info, llm_model_picked = 'Trust-Align-Qwen2.5'):
145
  chat_round = [
146
  {"role": "user",
 
141
  input = gr.Textbox(visible=False) # placeholder
142
  gr_md = gr.Markdown(mark_text + md_text_initial)
143
 
144
+ @spaces.GPU(duration=60)
145
  def update_with_rag_md(message, llm_results_use = 5, database_choice = index_info, llm_model_picked = 'Trust-Align-Qwen2.5'):
146
  chat_round = [
147
  {"role": "user",