Spaces:
Runtime error
Runtime error
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))
|