Rabbitt-AI commited on
Commit
4710ea4
·
verified ·
1 Parent(s): 8d9dd7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -329,7 +329,7 @@ def create_vector_db_and_annoy_index(pdf_path, vector_db_path, annoy_index_path)
329
  print("Vector database and Annoy index creation completed.")
330
 
331
 
332
- def chatbot_interface(file, user_query, response_style, extrapass):
333
  vector_db_path = "vector_db.pkl"
334
  annoy_index_path = "vector_index.ann"
335
  chunk_size = 2048
@@ -368,11 +368,10 @@ iface = gr.Interface(
368
  gr.File(label="Upload a PDF"),
369
  gr.Textbox(lines=5, label="User Query"),
370
  gr.Dropdown(["Detailed", "Concise", "Creative", "Technical"], label="Retreival Style"),
371
- gr.Dropdown(['Advanced Fusion'], value = "Advannced Fusion", label="Retrieval Method")
372
  ],
373
  outputs=gr.Textbox(label="RAG Response"),
374
  title="Chance RAG",
375
- description="Retrieval Method: Advanced Fusion"
376
  )
377
 
378
  iface.launch(share=True)
 
329
  print("Vector database and Annoy index creation completed.")
330
 
331
 
332
+ def chatbot_interface(file, user_query, response_style):
333
  vector_db_path = "vector_db.pkl"
334
  annoy_index_path = "vector_index.ann"
335
  chunk_size = 2048
 
368
  gr.File(label="Upload a PDF"),
369
  gr.Textbox(lines=5, label="User Query"),
370
  gr.Dropdown(["Detailed", "Concise", "Creative", "Technical"], label="Retreival Style"),
371
+ gr.Button("Retrieve with Advanced Fusion", variant="primary")
372
  ],
373
  outputs=gr.Textbox(label="RAG Response"),
374
  title="Chance RAG",
 
375
  )
376
 
377
  iface.launch(share=True)