Abbasid commited on
Commit
8a217b9
·
verified ·
1 Parent(s): ebd4b5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ class StoryEnd(Event):
37
 
38
  # Helper function to generate an image and return its path
39
  async def generate_image(prompt: str, hf_token: str) -> str | None:
40
- API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
41
  headers = {"Authorization": f"Bearer {hf_token}"}
42
  full_prompt = f"epic fantasy art, digital painting, cinematic lighting, masterpiece, {prompt}"
43
  payload = {"inputs": full_prompt}
 
37
 
38
  # Helper function to generate an image and return its path
39
  async def generate_image(prompt: str, hf_token: str) -> str | None:
40
+ API_URL = "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo"
41
  headers = {"Authorization": f"Bearer {hf_token}"}
42
  full_prompt = f"epic fantasy art, digital painting, cinematic lighting, masterpiece, {prompt}"
43
  payload = {"inputs": full_prompt}