Pijush2023 commited on
Commit
e52c5c7
·
verified ·
1 Parent(s): 1a4224a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -351,8 +351,8 @@ def generate_answer(message, choice, retrieval_mode):
351
  elif "restaurant" in message.lower() and "birmingham" in message.lower():
352
  intro = "Here are the top Restaurants in Birmingham:"
353
  response = fetch_yelp_restaurants()
354
- elif "flight" in message.lower() and ("JFK" in message or "BHM" in message):
355
- intro = "Here are some available flights for today:"
356
  response = fetch_google_flights()
357
  else:
358
  prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
351
  elif "restaurant" in message.lower() and "birmingham" in message.lower():
352
  intro = "Here are the top Restaurants in Birmingham:"
353
  response = fetch_yelp_restaurants()
354
+ elif "flight" in message.lower() and "birmingham" in message.lower():
355
+ # intro = "Here are some available flights for today:"
356
  response = fetch_google_flights()
357
  else:
358
  prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2