s

app.py

@@ -7,36 +7,20 @@ from dotenv import load_dotenv
 load_dotenv()
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-cascade-prior"
-# Corrected the format of the Authorization header here
-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)
-    # 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
+	return response.content
 
 prompt = st.text_input("Enter a prompt:", "batman hitting the griddy in gotham")
 
 image_bytes = query({
 	"inputs": prompt,
 })
-
-# 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.")
+st.text(image_bytes)
+# if(image_bytes):
+#     image = Image.open(io.BytesIO(image_bytes))
+#     st.image(image, caption="Image generated by the model", use_column_width=True)