Peng-Wang commited on
Commit
bc56d67
1 Parent(s): ae8872d

update bug

Browse files
Files changed (1) hide show
  1. caller.py +5 -10
caller.py CHANGED
@@ -70,11 +70,11 @@ class SeedT2ICaller():
70
  response = requests.post(
71
  t2i_url,
72
  headers=headers,
73
- data={
74
  'AlgoType': 1,
75
  'ReqJson': req_json,
76
  'AuthInfo': authInfo
77
- }
78
  )
79
  logger.info(f"header: {response.headers}")
80
  if response.status_code != 200:
@@ -88,7 +88,6 @@ class SeedT2ICaller():
88
  binary_data = binary_data1[0]
89
  #logger.info(f"binary_data: {binary_data}")
90
  image = Image.open(io.BytesIO(base64.b64decode(binary_data)))
91
- #image.save('./t2i_image.png')
92
  image = image.resize((self.cfg['resolution'], self.cfg['resolution']))
93
  return image, True
94
 
@@ -147,10 +146,6 @@ if __name__ == "__main__":
147
  "resolution": 611
148
  }
149
  model_t2i = SeedT2ICaller(cfg_t2i)
150
- model_t2i.generate("a beautiful girl")
151
-
152
- image_path = "./t2i_image.png"
153
- with open(image_path, 'rb') as image:
154
- image_bytes = image.read()
155
- model_edit = SeedEditCaller(cfg_t2i)
156
- model_edit.edit(image=image_bytes,edit="please edit to a good man")
 
70
  response = requests.post(
71
  t2i_url,
72
  headers=headers,
73
+ data=json.dumps({
74
  'AlgoType': 1,
75
  'ReqJson': req_json,
76
  'AuthInfo': authInfo
77
+ })
78
  )
79
  logger.info(f"header: {response.headers}")
80
  if response.status_code != 200:
 
88
  binary_data = binary_data1[0]
89
  #logger.info(f"binary_data: {binary_data}")
90
  image = Image.open(io.BytesIO(base64.b64decode(binary_data)))
 
91
  image = image.resize((self.cfg['resolution'], self.cfg['resolution']))
92
  return image, True
93
 
 
146
  "resolution": 611
147
  }
148
  model_t2i = SeedT2ICaller(cfg_t2i)
149
+ image, _ = model_t2i.generate("a beautiful girl")
150
+ model_edit = SeedEditCaller(cfg_t2i)
151
+ model_edit.edit(image, edit="please edit to a good man")