Update app.py

app.py

@@ -175,16 +175,16 @@ def query_summary(text):
     return text
 
 
-def query_generate(text, genres, year):
+def query_generate(text, title, genres, year):
     """
     Get image from HuggingFace Inference API
     -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/stabilityai/stable-diffusion-2-1"
+    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 + \
+    text = 'A Poster for the movie ' + title.split('(')[0] + 'in portrait mode based on the following synopsis: \"' + text + '\". Style: ' + genres + '. Year ' + year + \
     '. Ignore ' + ''.join(random.choices(string.ascii_letters, k=10))
     payload = {"inputs": f"{text}", "options": {"use_cache": "false"},}
     
@@ -215,6 +215,7 @@ def generate_poster(movie_data):
     genres_string = ', '.join(genres)
 
     year = movie_data['year'].values[0]
+    title = movie_data['title'].values[0]
 
 
     # Get summarization of movie synopsis
@@ -237,7 +238,7 @@ def generate_poster(movie_data):
 
     # Get image based on synopsis
     with st.spinner("Generating poster..."):
-        response_content = query_generate(synopsis_sum, genres_string, year)
+        response_content = query_generate(synopsis_sum, title, genres_string, year)
 
     # Show image
     try: