maialumiaho commited on
Commit
cc4f9e4
1 Parent(s): 4f94817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,10 +1,7 @@
1
  import streamlit as st
2
- import requests
3
- import io
4
 
5
  from PIL import Image
6
 
7
-
8
  #authorization
9
  API_URL = "https://api-inference.huggingface.co/models/xinsir/controlnet-union-sdxl-1.0"
10
 
@@ -20,4 +17,6 @@ def query(payload):
20
  def generate_image(prompt):
21
  image_bytes = query({"inputs": prompt})
22
  image = Image.open(io.BytesIO(image_bytes))
23
- return image
 
 
 
1
  import streamlit as st
 
 
2
 
3
  from PIL import Image
4
 
 
5
  #authorization
6
  API_URL = "https://api-inference.huggingface.co/models/xinsir/controlnet-union-sdxl-1.0"
7
 
 
17
  def generate_image(prompt):
18
  image_bytes = query({"inputs": prompt})
19
  image = Image.open(io.BytesIO(image_bytes))
20
+ return image
21
+
22
+