gaspar-avit commited on
Commit
79ef047
·
1 Parent(s): fe14d15

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -143,7 +143,7 @@ def load_dataset():
143
  ## ----------------------------- ##
144
 
145
  # Downloading Movies dataset
146
- api.dataset_download_file('rounakbanik/the-movies-dataset', 'movies_metadata.csv')
147
 
148
  # Extract data
149
  zf = ZipFile('movies_metadata.csv.zip')
@@ -244,6 +244,7 @@ def generate_poster(movie_data):
244
  st.text("")
245
  st.text("")
246
  st.subheader("Resulting poster:")
 
247
  col1, col2, col3 = st.columns([1, 5, 1])
248
  with col2:
249
  st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")
 
143
  ## ----------------------------- ##
144
 
145
  # Downloading Movies dataset
146
+ api.dataset_download_file('garymk/movielens-25m-dataset', 'movies.csv')
147
 
148
  # Extract data
149
  zf = ZipFile('movies_metadata.csv.zip')
 
244
  st.text("")
245
  st.text("")
246
  st.subheader("Resulting poster:")
247
+ st.text("")
248
  col1, col2, col3 = st.columns([1, 5, 1])
249
  with col2:
250
  st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")