Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,19 +60,6 @@ async def upload_image(image_file:UploadFile):
|
|
| 60 |
|
| 61 |
image_bytes = await image_file.read()
|
| 62 |
uploaded_image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
| 63 |
-
image_arr = np.array(uploaded_image)
|
| 64 |
-
|
| 65 |
-
gray_image = cv2.cvtColor(image_arr, cv2.COLOR_BGR2GRAY)
|
| 66 |
-
gray_image_arr = np.array(gray_image)
|
| 67 |
-
|
| 68 |
-
low_threshold = 100
|
| 69 |
-
high_threshold = 200
|
| 70 |
-
|
| 71 |
-
edges = cv2.Canny(gray_image_arr, low_threshold, high_threshold)
|
| 72 |
-
# Create an RGB image from the Canny edges
|
| 73 |
-
edges_rgb = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
|
| 74 |
-
|
| 75 |
-
uploaded_image = Image.fromarray(edges_rgb)
|
| 76 |
|
| 77 |
return JSONResponse(content={"message": "Logo uploaded successfully"})
|
| 78 |
|
|
@@ -139,6 +126,9 @@ async def generate_new_img(hex_code: str, prompt: str = Query(..., description="
|
|
| 139 |
image_bytes.seek(0)
|
| 140 |
init_image = Image.open(image_bytes).convert("RGB")
|
| 141 |
|
|
|
|
|
|
|
|
|
|
| 142 |
image = pipe(
|
| 143 |
ad_prompt, image=init_image
|
| 144 |
).images[0]
|
|
|
|
| 60 |
|
| 61 |
image_bytes = await image_file.read()
|
| 62 |
uploaded_image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
return JSONResponse(content={"message": "Logo uploaded successfully"})
|
| 65 |
|
|
|
|
| 126 |
image_bytes.seek(0)
|
| 127 |
init_image = Image.open(image_bytes).convert("RGB")
|
| 128 |
|
| 129 |
+
print(f"Image bytes length: {len(image_bytes.getvalue())}")
|
| 130 |
+
print(f"init_image type: {type(init_image)}")
|
| 131 |
+
|
| 132 |
image = pipe(
|
| 133 |
ad_prompt, image=init_image
|
| 134 |
).images[0]
|