dschandra commited on
Commit
3eef250
·
verified ·
1 Parent(s): df189db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import os
2
  import numpy as np
 
 
3
  import json
4
  import random
5
  from PIL import Image, ImageDraw, ImageFont
@@ -32,6 +34,10 @@ MODEL_MAP = {
32
  "AI Model Yifeng_3": 'models/yifeng_online/Yifeng_3.png',
33
  }
34
 
 
 
 
 
35
  def add_watermark(image):
36
  """
37
  Adds a watermark to the provided image.
 
1
  import os
2
  import numpy as np
3
+ import cv2
4
+ import base64
5
  import json
6
  import random
7
  from PIL import Image, ImageDraw, ImageFont
 
34
  "AI Model Yifeng_3": 'models/yifeng_online/Yifeng_3.png',
35
  }
36
 
37
+ def get_tryon_result(top_garment, bottom_garment):
38
+ # Your code to process garments...
39
+ encoded_top = base64.b64encode(cv2.imencode('.jpg', top_garment)[1].tobytes()).decode('utf-8')
40
+
41
  def add_watermark(image):
42
  """
43
  Adds a watermark to the provided image.