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

Update app.py

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