Update app.py

app.py

@@ -181,7 +181,8 @@ def query_generate(text, genres, year):
     -param text: text to generate image
     -return: generated image
     """
-    API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+    #API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
+    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
     headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
     text = 'A movie Poster in portrait mode based on the following synopsis: \"' + text + '\". Style: ' + genres + ', year ' + year + \
     '. Ignore ' + ''.join(random.choices(string.ascii_letters, k=10))
@@ -287,6 +288,12 @@ if __name__ == "__main__":
 
     # Set page footer
     footer()
+
+    # Set sidebar with info
+    st.sidebar.markdown("## Generating movie posters using Stable Diffusion")
+    st.sidebar.markdown("This streamlit space aims to generate movie posters based on synopsis.")
+    st.sidebar.markdown("Firstly, the synopsis of the selected movie is extracted from the dataset and then summarized using Facebook's BART model.")
+    st.sidebar.markdown("Once the movie's summary is ready, it is passed to the Stable Diffusion v1.5 model using HF's Inference API, with some prompt tuning.")
     ## ---------------------------- ##