syurein commited on
Commit
941164e
·
1 Parent(s): 3ac0ed9

json.loadsがなかっただけだった

Browse files
Files changed (1) hide show
  1. LLM_package.py +2 -2
LLM_package.py CHANGED
@@ -102,7 +102,7 @@ class ObjectDetector:
102
 
103
  response = self.model.parse(self.model.get_response(image_path, analysis_prompt))
104
  print(f"Response: {response}")
105
- return response
106
  def detect_danger_level(self, image_path):
107
  analysis_prompt = f"""
108
  画像の個人情報漏洩リスクを分析し、厳密にJSON形式で返答してください。なおこの時、資料があれば、資料を参考にしてください:
@@ -113,7 +113,7 @@ class ObjectDetector:
113
  {self.text if self.text else "なし"}
114
  </資料>
115
  """
116
- response = json.loads(self.model.parse(self.model.get_response(image_path, analysis_prompt)))
117
 
118
  print(f"Response: {response}")
119
  try:
 
102
 
103
  response = self.model.parse(self.model.get_response(image_path, analysis_prompt))
104
  print(f"Response: {response}")
105
+ return json.loads(response)
106
  def detect_danger_level(self, image_path):
107
  analysis_prompt = f"""
108
  画像の個人情報漏洩リスクを分析し、厳密にJSON形式で返答してください。なおこの時、資料があれば、資料を参考にしてください:
 
113
  {self.text if self.text else "なし"}
114
  </資料>
115
  """
116
+ response = self.model.parse(self.model.get_response(image_path, analysis_prompt))
117
 
118
  print(f"Response: {response}")
119
  try: