修改app.py

This commit is contained in:
lee
2025-01-22 13:25:05 +08:00
parent c9d79f8059
commit ff0ac412f8
2 changed files with 7 additions and 5 deletions

10
app.py
View File

@ -47,8 +47,10 @@ def item_analysis(stream_dict):
if stream_dict['goodsPic'] is not None:
response = requests.get(stream_dict['goodsPic'])
std_img = Image.open(BytesIO(response.content))
description_dict = get_product_description(std_img, track_imgs, initModel.qwen_model, initModel.processor)
print(description_dict)
description_dict = get_product_description(std_img,
track_imgs,
initModel.qwen_model,
initModel.processor)
return description_dict
@ -77,10 +79,10 @@ def aiReview(): # put application's code here
data[key] = None
logging.error(f'{key} is null')
# stream_pipeline(data)
item_analysis(data)
description_dict = item_analysis(data)
end = time.time()
logging.info(f'aiReview cost {end - start}s')
return 0
return description_dict
# def main():

View File

@ -16,7 +16,7 @@ def aiReviewClient():
}
url = "http://192.168.1.28:8060/ai_review"
r = requests.post(url=url, data=json.dumps(data))
print(r.text)
print(r.json())
if __name__ == '__main__':