Upload 2 files

app.py

@@ -185,6 +185,9 @@ def generate_poster(movie_data):
     # Get summarization of movie synopsis
     with st.spinner("Please wait while the synopsis is being summarized..."):
         synopsis_sum = query_summarization(movie_data.overview.values[0])
+
+    st.text("")
+    st.text("")
     st.text(synopsis_sum)
 
     # Load text-to-image model
@@ -193,12 +196,11 @@ def generate_poster(movie_data):
             pipeline = load_model()
     
     # Get image based on synopsis
-    image = pipeline(prompt).images[0]
-    st.image(image, caption=movie_data.title)
-
-    return None #poster_image
-
+    poster_image = pipeline(synopsis_sum).images[0]
+    st.image(poster_image, caption=movie_data.title)
 
+    return poster_image
+# ------------------------------------------------------- #
 
 
 ###############################
@@ -251,5 +253,5 @@ if __name__ == "__main__":
     st.text("")
     st.text("")
 
-    #if image is not None:
-    #    st.image(image, caption=session.selected_movie.title.values[0])
+    if image is not None:
+        st.image(image, caption=session.selected_movie.title.values[0])

requirements.txt

@@ -3,4 +3,5 @@ kaggle
 zipfile39
 sentence-transformers
 diffusers
-accelerate
+accelerate
+htbuilder