update

app.py

@@ -1,4 +1,3 @@
-# Python
 import streamlit as st
 import requests
 import io
@@ -8,13 +7,20 @@ 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')}"}
+# Corrected the format of the Authorization header here
+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)
-	return response.content
+    # Check if the request was successful
+    if response.status_code == 200:
+        return response.content
+    else:
+        # If the request failed, print the response and return None
+        st.write("Failed to get the image:", response.text)
+        return None
 
 prompt = st.text_input("Enter a prompt:", "batman hitting the griddy in gotham")
 
@@ -22,6 +28,15 @@ image_bytes = query({
 	"inputs": prompt,
 })
 
-if(image_bytes):
-    image = Image.open(io.BytesIO(image_bytes))
-    st.image(image, caption="Image generated by the model", use_column_width=True)
+# Ensure that image_bytes is not None and is not empty
+if image_bytes:
+    try:
+        image = Image.open(io.BytesIO(image_bytes))
+        st.image(image, caption="Image generated by the model", use_column_width=True)
+    except IOError:
+        # If an error occurs, print the error and the image_bytes
+        st.write("Cannot identify image file. The response may not be an image or is corrupted.")
+        # To debug, you might want to see the first few bytes to determine if it's an image file
+        st.write(image_bytes[:100])
+else:
+    st.write("No image data was returned.")