Mustafaansari commited on
Commit
74c189b
·
verified ·
1 Parent(s): a6a5695

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -1,16 +1,27 @@
1
  import modelbit
2
  import pandas as pd
3
  import gradio as gr
 
4
  def recommend(movie_name):
5
  response_json = modelbit.get_inference(
6
  region="ap-south-1",
7
  workspace="mustafaansari",
8
  deployment="recommend",
9
- data= movie_name
10
  )
11
  data_list = response_json['data']
12
  df = pd.DataFrame(data_list)
13
- return df
14
- gr.Interface(fn=recommend, inputs="text", outputs="dataframe",title="Hollywood Movie Recommendation System",
15
- description="Enter a Hollywood movie name, and this system will recommend similar movies based on your input."
16
- ).launch()
 
 
 
 
 
 
 
 
 
 
 
1
  import modelbit
2
  import pandas as pd
3
  import gradio as gr
4
+
5
  def recommend(movie_name):
6
  response_json = modelbit.get_inference(
7
  region="ap-south-1",
8
  workspace="mustafaansari",
9
  deployment="recommend",
10
+ data=movie_name
11
  )
12
  data_list = response_json['data']
13
  df = pd.DataFrame(data_list)
14
+ return df.to_dict(orient='records')
15
+
16
+ # Add an example for the UI
17
+ examples = [
18
+ ["The Dark Knight"],
19
+ ["Inception"],
20
+ ["Interstellar"],
21
+ ["Aliens vs Predator: Requiem"]
22
+ ]
23
+
24
+ gr.Interface(fn=recommend, inputs="text", outputs="json", title="Hollywood Movie Recommendation System",
25
+ description="Enter a Hollywood movie name, and this system will recommend similar movies based on your input.",
26
+ examples=examples # Add examples to the UI
27
+ ).launch()