Kaushik066 commited on
Commit
0839849
·
verified ·
1 Parent(s): b84683e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -7
app.py CHANGED
@@ -32,11 +32,12 @@ else:
32
 
33
  def respond(
34
  message,
35
- #history: list[tuple[str, str]],
36
- #system_message,
37
  max_tokens,
38
- temperature,
39
- top_p):
 
40
 
41
  # define retriever object
42
  retriever = vectordb.as_retriever(search_type="similarity", search_kwargs={"k": 5})
@@ -50,15 +51,17 @@ def respond(
50
  # create a RAG pipeline
51
  qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever)
52
  #generate results
53
- result = qa_chain.invoke(query)
54
 
55
  yield result['result']
56
 
57
 
58
  demo = gr.ChatInterface(
59
  respond,
 
 
60
  additional_inputs=[
61
- #gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
62
  gr.Slider(minimum=128, maximum=1024, value=512, step=128, label="Max new tokens"),
63
  gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
64
  gr.Slider(
@@ -66,7 +69,7 @@ demo = gr.ChatInterface(
66
  maximum=1.0,
67
  value=0.95,
68
  step=0.05,
69
- #label="Top-p (nucleus sampling)",
70
  ),
71
  ],
72
  )
 
32
 
33
  def respond(
34
  message,
35
+ history: list[tuple[str, str]],
36
+ system_message,
37
  max_tokens,
38
+ temperature
39
+ #top_p
40
+ ):
41
 
42
  # define retriever object
43
  retriever = vectordb.as_retriever(search_type="similarity", search_kwargs={"k": 5})
 
51
  # create a RAG pipeline
52
  qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever)
53
  #generate results
54
+ result = qa_chain.invoke(message)
55
 
56
  yield result['result']
57
 
58
 
59
  demo = gr.ChatInterface(
60
  respond,
61
+ type="messages",
62
+ autofocus=False,
63
  additional_inputs=[
64
+ gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
65
  gr.Slider(minimum=128, maximum=1024, value=512, step=128, label="Max new tokens"),
66
  gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature"),
67
  gr.Slider(
 
69
  maximum=1.0,
70
  value=0.95,
71
  step=0.05,
72
+ label="Top-p (nucleus sampling)",
73
  ),
74
  ],
75
  )