farukbera commited on
Commit
d894182
·
1 Parent(s): bf9e626

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
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]