Mridul2003 commited on
Commit
92449cf
·
verified ·
1 Parent(s): a0ffacf

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -0
api.py CHANGED
@@ -122,6 +122,8 @@ def filter_comment():
122
  image_bytes = image_file.read()
123
  image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
124
  result = image_classifier.classify(image)
 
 
125
  logger.info("Image classification result: %s", result)
126
  return jsonify({"image_classification": result})
127
  except Exception as e:
 
122
  image_bytes = image_file.read()
123
  image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
124
  result = image_classifier.classify(image)
125
+ if result.get("text"):
126
+ result['toxic_result'] = text_filter_service.process_text(result.get("text"))
127
  logger.info("Image classification result: %s", result)
128
  return jsonify({"image_classification": result})
129
  except Exception as e: