s

app.py

@@ -8,17 +8,17 @@ from dotenv import load_dotenv
 load_dotenv()
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-cascade-prior"
-headers = {"Authorization": f"Bearer ${os.getenv('bearer_token')}"}
+HEADERS = {"Authorization": f"Bearer ${os.getenv('bearer_token')}"}
 
 st.title("Image Generator with Diffusers")
 
-def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
+def async query(payload):
+	response = requests.post(API_URL, headers=HEADERS, json=payload)
 	return response.content
 
 prompt = st.text_input("Enter a prompt:", "batman hitting the griddy in gotham")
 
-image_bytes = query({
+image_bytes = await query({
 	"inputs": prompt,
 })