File size: 489 Bytes
6459725
995f54d
 
 
5f60760
6459725
5f60760
 
6459725
995f54d
5f60760
6459725
 
 
d1b37d6
6459725
 
d1b37d6
 
 
5f60760
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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()