Update app.py

app.py

@@ -16,11 +16,14 @@ translator = pipeline("translation", model="Helsinki-NLP/opus-mt-dra-en")
 summarizer = pipeline("summarization", model="Falconsai/text_summarization")
 # image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16).to("cpu")
 
-API_URL = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
-headers = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
+
+
+# for image api
+IMAGE_API_URL = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
+img_headers = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
 
 def query(payload):
-	response = requests.post(API_URL, headers=headers, json=payload)
+	response = requests.post(IMAGE_API_URL, headers=img_headers, json=payload)
 	return response.content