Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,26 +11,20 @@ hf_token = os.environ.get("hf_token")
|
|
11 |
|
12 |
# Function to generate a QR code with transparent fill color
|
13 |
def generate_qr_image(url, image_size):
|
14 |
-
# Determine box size based on the larger dimension of the image
|
15 |
-
max_dim = max(image_size[0], image_size[1])
|
16 |
-
box_size = max_dim // 25
|
17 |
-
border = 1
|
18 |
-
|
19 |
qr = qrcode.QRCode(
|
20 |
version=1,
|
21 |
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
22 |
-
box_size=
|
23 |
-
border=
|
24 |
)
|
25 |
qr.add_data(url)
|
26 |
qr.make(fit=True)
|
27 |
|
28 |
-
qr_img = qr.make_image(fill_color=
|
29 |
qr_img = qr_img.resize(image_size)
|
30 |
|
31 |
return qr_img
|
32 |
|
33 |
-
|
34 |
# Function to query the image from the Hugging Face API
|
35 |
def query_image(text):
|
36 |
API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
|
@@ -46,7 +40,7 @@ def generate_image(url, text):
|
|
46 |
api_image = Image.open(io.BytesIO(image_bytes))
|
47 |
|
48 |
# Generate QR code image
|
49 |
-
qr_image = generate_qr_image(url,
|
50 |
|
51 |
# Create a blank image with transparency
|
52 |
final_image = Image.new('RGBA', api_image.size, (255, 255, 255, 0))
|
|
|
11 |
|
12 |
# Function to generate a QR code with transparent fill color
|
13 |
def generate_qr_image(url, image_size):
|
|
|
|
|
|
|
|
|
|
|
14 |
qr = qrcode.QRCode(
|
15 |
version=1,
|
16 |
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
17 |
+
box_size=50,
|
18 |
+
border=4,
|
19 |
)
|
20 |
qr.add_data(url)
|
21 |
qr.make(fit=True)
|
22 |
|
23 |
+
qr_img = qr.make_image(fill_color=None, back_color="white").convert("RGBA")
|
24 |
qr_img = qr_img.resize(image_size)
|
25 |
|
26 |
return qr_img
|
27 |
|
|
|
28 |
# Function to query the image from the Hugging Face API
|
29 |
def query_image(text):
|
30 |
API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
|
|
|
40 |
api_image = Image.open(io.BytesIO(image_bytes))
|
41 |
|
42 |
# Generate QR code image
|
43 |
+
qr_image = generate_qr_image(url, (250, 250))
|
44 |
|
45 |
# Create a blank image with transparency
|
46 |
final_image = Image.new('RGBA', api_image.size, (255, 255, 255, 0))
|