gaspar-avit commited on
Commit
ff60cfc
·
1 Parent(s): e4d0f4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -182,8 +182,8 @@ def query_generate(text, title, genres, year):
182
  -return: generated image
183
  """
184
 
185
- API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
186
- #API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
187
  #API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
188
  headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
189
  text = 'A Poster for the movie ' + title.split('(')[0] + 'in portrait mode based on the following synopsis: \"' + text + '\". Style: ' + genres + '. Year ' + year + \
 
182
  -return: generated image
183
  """
184
 
185
+ #API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
186
+ API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
187
  #API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2-1"
188
  headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
189
  text = 'A Poster for the movie ' + title.split('(')[0] + 'in portrait mode based on the following synopsis: \"' + text + '\". Style: ' + genres + '. Year ' + year + \