syurein commited on
Commit
bae4865
·
1 Parent(s): a345b5c

エラーハンドリングの強化

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1015,7 +1015,7 @@ async def classify_image_llm2(file: UploadFile = File(...), api_key: str = None)
1015
  buffer.write(await file.read())
1016
  if not api_key:
1017
  api_key = os.getenv('GEMINI_API_KEY')
1018
- danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
1019
  return {"danger":danger_level}
1020
 
1021
 
@@ -1036,7 +1036,7 @@ async def classify_image_llm(file: UploadFile = File(...), api_key: str = None):
1036
  buffer.write(await file.read())
1037
  if not api_key:
1038
  api_key = os.getenv('GEMINI_API_KEY')
1039
- danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
1040
  return {"danger":danger_level}
1041
 
1042
  '''
 
1015
  buffer.write(await file.read())
1016
  if not api_key:
1017
  api_key = os.getenv('GEMINI_API_KEY')
1018
+ danger_level = ObjectDetector(api_key=api_key).detect_danger_level(image_path)
1019
  return {"danger":danger_level}
1020
 
1021
 
 
1036
  buffer.write(await file.read())
1037
  if not api_key:
1038
  api_key = os.getenv('GEMINI_API_KEY')
1039
+ danger_level = ObjectDetector(api_key=api_key).detect_danger_level(image_path)
1040
  return {"danger":danger_level}
1041
 
1042
  '''