Danielrahmai1991 commited on
Commit
b183ef0
1 Parent(s): 57b94d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,7 +96,7 @@ print("retriver done")
96
  def greet(question, model_type):
97
 
98
  print(f"question is {question}")
99
- if model_type == "With memory":
100
  out_gen = rag_chain.invoke(question)
101
  print("test5")
102
  print(f"out is: {out_gen}")
@@ -108,6 +108,6 @@ def greet(question, model_type):
108
  return out_gen
109
 
110
  demo = gr.Interface(fn=greet, inputs=["text", gr.Dropdown(
111
- ["Without memory", "With memory"], label="Memory status", info="With using memory, the output will be slow but strong"
112
  ),], outputs="text")
113
  demo.launch(debug=True, share=True)
 
96
  def greet(question, model_type):
97
 
98
  print(f"question is {question}")
99
+ if model_type == "With RAG":
100
  out_gen = rag_chain.invoke(question)
101
  print("test5")
102
  print(f"out is: {out_gen}")
 
108
  return out_gen
109
 
110
  demo = gr.Interface(fn=greet, inputs=["text", gr.Dropdown(
111
+ ["Without RAG", "With RAG"], label="Memory status", info="With using memory, the output will be slow but strong"
112
  ),], outputs="text")
113
  demo.launch(debug=True, share=True)