raghuv-aditya commited on
Commit
d683b77
·
verified ·
1 Parent(s): fe3de43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,12 +15,12 @@ def main(query):
15
  Returns:
16
  str: Response from the chatbot with a recommended course.
17
  """
18
- courses_data = scrape_courses_json(BASE_URL, num_pages=1)
19
  course_text = generate_text(courses_data)
20
  vector_store = process_safety_with_chroma(course_text)
21
  qa_system = create_chatbot(vector_store)
22
 
23
- prompt = "Suggest me the best course for " + query + " in a structured format with a link."
24
  return ask_question(qa_system, prompt)
25
 
26
  # Gradio interface
 
15
  Returns:
16
  str: Response from the chatbot with a recommended course.
17
  """
18
+ courses_data = scrape_courses_json(BASE_URL, num_pages=8)
19
  course_text = generate_text(courses_data)
20
  vector_store = process_safety_with_chroma(course_text)
21
  qa_system = create_chatbot(vector_store)
22
 
23
+ prompt = "Suggest me the best course for " + query + " in a structured format with a link from the content provided only."
24
  return ask_question(qa_system, prompt)
25
 
26
  # Gradio interface