Upload 2 files

AIMAGE.exe

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9fd8ab82cfea6367852a91068021b9799db42718e160eeb89e8c1e82d09b23c8
-size 36301732
+oid sha256:acb94b451549526fcf57b208b0f5ab9b6cd3cfcae71cc75b10342535a99ad754
+size 38548043

AIMAGE.py

@@ -1,12 +1,21 @@
-from requests import post
-from io import BytesIO
-from PIL.Image import open as open_image
-from tkinter.simpledialog import askstring
-
-while True:
-    text = askstring("IMAGE PROMPT",prompt="What is your prompt?")
-    if text and text.strip() != "":
-        hfapi = "<KEY>"
-        open_image(BytesIO(post("https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0", headers={"Authorization": f"Bearer {hfapi}"}, json={"inputs": text.strip()}).content)).show()
-    else:
-        break
+try:
+    from requests import post
+    from io import BytesIO
+    from PIL.Image import open as open_image
+    from tkinter.simpledialog import askstring
+    from tkinter.messagebox import showerror
+    hfapi = "hf_token"
+    while True:
+        text = askstring("IMAGE PROMPT",prompt="What is your prompt?")
+        if text and text.strip() != "":
+            postreq = post("https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0", headers={"Authorization": f"Bearer {hfapi}"}, json={"inputs": text.strip()})
+            if postreq.status_code == 200:
+                open_image(BytesIO(postreq.content)).show()
+            else:
+                showerror("ERROR",f"Error: Software is out of service or your internet connection is not working.")
+                break
+        else:
+            break
+except Exception:
+    showerror("ERROR",f"Error: Software is out of service or your internet connection is not working.")
+