Upload app.py

app.py

@@ -186,7 +186,9 @@ def query_generate(text):
     API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
     headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
     text = "Poster of movie. " + text
-    payload = {"inputs": f"{text}",}
+    payload = {"inputs": f"{text}",
+               "options": {"wait_for_model": True},
+              }
     
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
@@ -201,10 +203,10 @@ def generate_poster(movie_data):
 
 
     # Get summarization of movie synopsis
+    st.text("")
     with st.spinner("Please wait while the synopsis is being summarized..."):
         synopsis_sum = query_summary(movie_data.overview.values[0])
 
-    st.text("")
     st.text("")
     st.subheader("Synopsis:")
     st.text("Synopsis summary: " + synopsis_sum)
@@ -223,13 +225,12 @@ def generate_poster(movie_data):
         st.text("")
         st.text("")
         st.subheader("Resulting poster:")
-        col1, col2, col3 = st.columns([1, 10, 1])
+        col1, col2, col3 = st.columns([1, 6, 1])
 
         with col1:
             st.write("")
 
         with col2:
-            st.text("")
             st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")
 
         with col3:
@@ -278,7 +279,7 @@ if __name__ == "__main__":
     st.text("")
 
     ## Select box with all the movies as choices
-    selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title_year)
+    session.selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title_year)
 
     st.text("")
     st.text("")
@@ -289,7 +290,7 @@ if __name__ == "__main__":
 
     ## Generate poster
     if is_clicked:
-        poster = generate_poster(data[data.title_year==selected_movie])
+        poster = generate_poster(data[data.title_year==session.selected_movie])
         generate_poster.clear()
         st.runtime.legacy_caching.clear_cache()