Create app.py

app.py

@@ -0,0 +1,31 @@
+import streamlit as st
+import requests
+import io
+from PIL import Image
+import os
+
+hf_token = os.environ.get("spaces_prototype")
+API_URL_SD = "https://api-inference.huggingface.co/models/xinsir/controlnet-union-sdxl-1.0"
+
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+
+st.title("text to image model")
+st.write("Enter a prompt")
+model = st.selectbox(
+  "choose model",
+  ("stable diffusion")
+
+)
+prompt = st.text_input("Enter prompt")
+
+if prompt:
+  if model == "stable diffusion":
+    API_URL = API_URL_SD
+  image_bytes = query({
+	"inputs": prompt,
+  })
+  st.image(image_bytes, caption="Generated image")
+  st.info("Image generated successfully")
+