gaspar-avit commited on
Commit
0984e5d
·
1 Parent(s): 3291051

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -187,7 +187,9 @@ def query_generate(text, genres, year):
187
  """
188
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
189
  headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
190
- text = 'A movie Poster based on the following synopsis: \"' + text + '\". Style: ' + genres + ', year ' + year
 
 
191
  payload = {"inputs": f"{text}",}
192
 
193
  response = requests.post(API_URL, headers=headers, json=payload)
@@ -312,9 +314,8 @@ if __name__ == "__main__":
312
  ## Generate poster
313
  if is_clicked:
314
  poster = generate_poster(data[data.title_year==session.selected_movie])
315
- st.runtime.legacy_caching.clear_cache()
316
  generate_poster.clear()
317
- is_clicked = False
318
 
319
 
320
  _= """
 
187
  """
188
  API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
189
  headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
190
+ text = 'A movie Poster based on the following synopsis: \"' + text + '\". Style: ' + genres + ', year ' + year + \
191
+ '. Ignore ' + ''.join(random.choices(string.ascii_letters, k=10))
192
+ st.write(text)
193
  payload = {"inputs": f"{text}",}
194
 
195
  response = requests.post(API_URL, headers=headers, json=payload)
 
314
  ## Generate poster
315
  if is_clicked:
316
  poster = generate_poster(data[data.title_year==session.selected_movie])
 
317
  generate_poster.clear()
318
+ st.runtime.legacy_caching.clear_cache()
319
 
320
 
321
  _= """