性能向上

LLM_package.py

@@ -75,8 +75,8 @@ class GeminiInference:
                 parsed.append({'label': obj['label'], 'box_2d': norm})
         return parsed
 class ObjectDetector:
-    def __init__(self, api_key=os.getenv('GEMINI_API_KEY')): # API_KEYをapi_keyに変更し、デフォルト値を設定
-        self.model  = GeminiInference(api_key) # ここもapi_keyを使用
+    def __init__(self, API_KEY=os.getenv('GEMINI_API_KEY')): # API_KEYをapi_keyに変更し、デフォルト値を設定
+        self.model  = GeminiInference(API_KEY) # ここもapi_keyを使用
         self.prompt_objects=None
         self.text=None
         self.scene=None

app.py

@@ -1015,7 +1015,7 @@ async def classify_image_llm2(file: UploadFile = File(...), api_key: str = None)
         buffer.write(await file.read())
     if not api_key:
         api_key = os.getenv('GEMINI_API_KEY')
-    danger_level = ObjectDetector(api_key=api_key).detect_danger_level(image_path)
+    danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
     return {"danger":danger_level}
 
 
@@ -1036,7 +1036,7 @@ async def classify_image_llm(file: UploadFile = File(...), api_key: str = None):
         buffer.write(await file.read())
     if not api_key:
         api_key = os.getenv('GEMINI_API_KEY')
-    danger_level = ObjectDetector(api_key=api_key).detect_danger_level(image_path)
+    danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
     return {"danger":danger_level}
 
 '''