capradeepgujaran commited on
Commit
c498e95
·
verified ·
1 Parent(s): 7870fce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ class RobustSafetyMonitor:
94
  stream=False
95
  )
96
  # Process and parse the response correctly
97
- response = completion.choices[0].message.content
98
  return self.parse_safety_analysis(response), response # Return parsed analysis and full response
99
 
100
  except Exception as e:
 
94
  stream=False
95
  )
96
  # Process and parse the response correctly
97
+ response = completion.choices[0].message['content']
98
  return self.parse_safety_analysis(response), response # Return parsed analysis and full response
99
 
100
  except Exception as e: