Upload app.py

app.py

@@ -189,6 +189,7 @@ def query_generate(text, genres, year):
     headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
     text = 'A movie Poster based on the following synopsis: \"' + text + '\". Style: ' + genres + ', year ' + year + \
     '. Ignore ' + ''.join(random.choices(string.ascii_letters, k=10))
+    st.write(text)
     payload = {"inputs": f"{text}",}
     
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -237,7 +238,7 @@ def generate_poster(movie_data):
 
 
     # Get image based on synopsis
-    with st.spinner("Generating poster image..."):
+    with st.spinner("Generating poster image... This could take a few minutes."):
         response_content = query_generate(synopsis_sum, genres_string, year)
 
     # Show image
@@ -306,15 +307,15 @@ if __name__ == "__main__":
     buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
     is_clicked = col1.button(label="Generate poster!")
 
+    ## Clear cache between runs
+    st.runtime.legacy_caching.clear_cache()
+    generate_poster.clear()
+
     ## Generate poster
     if is_clicked:
         poster = generate_poster(data[data.title_year==session.selected_movie])
-        #generate_poster.clear()
-        #st.runtime.legacy_caching.clear_cache()
-
-    ## Clear cache between runs
-    #st.runtime.legacy_caching.clear_cache()
-    #generate_poster.clear()
+        generate_poster.clear()
+        st.runtime.legacy_caching.clear_cache()
 
 
     _= """