Update app.py

app.py

@@ -4,7 +4,8 @@ import io
 import requests
 import os 
 HF_TOKEN = os.environ["HF_TOKEN"]
-
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.content
@@ -21,8 +22,7 @@ def main():
         # Call image_generator function
         
 
-        API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+        
         image_bytes = query({
 	"inputs": text_input,
 })