gaspar-avit commited on
Commit
09e6886
·
1 Parent(s): d97d032

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -185,6 +185,8 @@ def query_summarization(text):
185
  payload = {"inputs": f"{text}",}
186
 
187
  response = requests.request("POST", API_URL, headers=headers, json=payload).json()
 
 
188
  return response.get('summary_text')
189
 
190
  def generate_poster(movie_data):
@@ -194,8 +196,6 @@ def generate_poster(movie_data):
194
  -return: image of generated alternative poster
195
  """
196
 
197
- st.text(movie_data.overview.values[0])
198
- st.text(movie_data.overview[0])
199
 
200
  # Get summarization of movie synopsis
201
  with st.spinner("Please wait while the synopsis is being summarized..."):
@@ -233,7 +233,10 @@ if __name__ == "__main__":
233
  # Set page title
234
  st.title("""
235
  Movie Poster Generator :film_frames:
236
- This is a movie poster generator based on movie's synopsis :sunglasses: \nJust select the title of a movie to generate an alternative poster.
 
 
 
237
  """)
238
 
239
  # Set page footer
 
185
  payload = {"inputs": f"{text}",}
186
 
187
  response = requests.request("POST", API_URL, headers=headers, json=payload).json()
188
+ st.text(response)
189
+
190
  return response.get('summary_text')
191
 
192
  def generate_poster(movie_data):
 
196
  -return: image of generated alternative poster
197
  """
198
 
 
 
199
 
200
  # Get summarization of movie synopsis
201
  with st.spinner("Please wait while the synopsis is being summarized..."):
 
233
  # Set page title
234
  st.title("""
235
  Movie Poster Generator :film_frames:
236
+
237
+ This is a movie poster generator based on movie's synopsis :sunglasses:
238
+
239
+ Just select the title of a movie to generate an alternative poster.
240
  """)
241
 
242
  # Set page footer