Quardo commited on
Commit
9ee97ba
1 Parent(s): 5135341

Updated Space.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -497,7 +497,7 @@ async def respond(
497
  async for token in completion:
498
  response += token['choices'][0]['delta'].get("content", token['choices'][0]['delta'].get("refusal", ""))
499
  yield f"{prefix}{response}"
500
- mode = moderate([{"role":"user","content":user_message},{"role": "assistant", "content": response}])
501
  if mode:
502
  reasons = []
503
  categories = mode[0].get('categories', {}) if isinstance(mode, list) else mode.get('categories', {})
 
497
  async for token in completion:
498
  response += token['choices'][0]['delta'].get("content", token['choices'][0]['delta'].get("refusal", ""))
499
  yield f"{prefix}{response}"
500
+ mode = moderate([handleMultimodalData(model_name, "user", message),{"role": "assistant", "content": response}])
501
  if mode:
502
  reasons = []
503
  categories = mode[0].get('categories', {}) if isinstance(mode, list) else mode.get('categories', {})