Tom Beer
close loop
995f54d
raw
history blame
No virus
489 Bytes
from gradio import Interface, Dropdown, CheckboxGroup
from data import get_cities
from llm import get_response
cities = get_cities()
def hotel_recommender(city, preferences) -> dict:
return get_response(city, preferences)
demo = Interface(
fn=hotel_recommender,
inputs=[
Dropdown(cities, label="City"),
CheckboxGroup(["Kid friendly", "other option"], label="Preferences"),
],
outputs="text"
)
if __name__ == "__main__":
demo.launch()