Soumalya Das commited on
Commit
9012408
·
verified ·
1 Parent(s): b0986f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -22
app.py CHANGED
@@ -1,23 +1,23 @@
1
- import gradio as gr
2
- from api import MovieRecommender
3
-
4
- recommender = MovieRecommender()
5
-
6
- def recommend_movies(prompt, topk):
7
- df = recommender.recommend(prompt, topk=int(topk))
8
- return df
9
-
10
- demo = gr.Interface(
11
- fn=recommend_movies,
12
- inputs=[
13
- gr.Textbox(label="Movie prompt", placeholder="action thriller with robots"),
14
- gr.Slider(1, 20, value=5, step=1, label="Top K")
15
- ],
16
- outputs=gr.Dataframe(label="Recommendations"),
17
- title="🎬 Movie Nerd",
18
- description="Prompt-based movie recommendations using embeddings"
19
- )
20
-
21
- if __name__ == "__main__":
22
- demo.launch()
23
 
 
1
+ import gradio as gr
2
+ from api import MovieRecommender
3
+
4
+ recommender = MovieRecommender()
5
+
6
+ def recommend_movies(prompt, topk):
7
+ df = recommender.recommend(prompt, topk=int(topk))
8
+ return df
9
+
10
+ demo = gr.Interface(
11
+ fn=recommend_movies,
12
+ inputs=[
13
+ gr.Textbox(label="Movie prompt", placeholder="action thriller with robots"),
14
+ gr.Slider(1, 20, value=5, step=1, label="Top K")
15
+ ],
16
+ outputs=gr.Dataframe(label="Recommendations"),
17
+ title="🎬 Movie Nerd",
18
+ description="Prompt-based movie recommendations using embeddings"
19
+ )
20
+
21
+ if __name__ == "__main__":
22
+ demo.launch(share=True)
23