Upload app.py

app.py

@@ -219,20 +219,22 @@ def generate_poster(movie_data):
     st.text("")
     st.text("Synopsis summary: " + synopsis_sum)
 
-    # 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]
-        poster_image = query_gen(synopsis_sum)
-
-    import io
-    from PIL import Image
-    image = Image.open(io.BytesIO(poster_image))
-    st.image(image)
+    if 1:
+        # 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]
+            poster_image = query_gen(synopsis_sum)
+
+    else:
+        # 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)
 
     return poster_image
@@ -292,4 +294,4 @@ if __name__ == "__main__":
     st.text("")
 
     if image is not None:
-        st.image(image, caption=session.selected_movie.title.values[0])
+        st.image(image, caption=session.selected_movie.title)