Upload app.py

app.py

@@ -162,6 +162,7 @@ def load_dataset():
 
     # Create dataframe
     data = pd.read_csv('movies_metadata.csv', low_memory=False)
+    data.drop_duplicates(inplace=True)
 
     return data
 
@@ -230,21 +231,13 @@ def generate_poster(movie_data):
 
     import io
     from PIL import Image
-    image = Image.open(io.BytesIO(poster_image))
-    st.image(image, caption=movie_data.title)
-
-    '''
-    # Load text-to-image model
-    if not IS_MODEL_LOADED:
-        with st.spinner("Loading Text to Image model..."):
-            pipeline = load_model()
-    
-    # Get image based on synopsis
-    with st.spinner("Generating poster image...\nThis may take some time, please be patient"):
-        poster_image = pipeline(synopsis_sum).images[0]
-
-    st.image(poster_image, caption=movie_data.title)
-    '''
+    try: 
+        image = Image.open(io.BytesIO(poster_image))
+        st.image(image, caption=movie_data.title.values[0])
+        return image
+    except:
+        st.text(poster_image)
+        return poster_image
 
     return image
 # ------------------------------------------------------- #
@@ -302,4 +295,4 @@ if __name__ == "__main__":
     st.text("")
 
     if image is not None:
-        st.image(image, caption=session.selected_movie.title)
+        st.image(image, caption=session.selected_movie.title.values[0])