Spaces:

cosmin1769
/

CalorieTrackerAI-backend

Sleeping

cosmin1769 commited on 16 days ago

Commit

69d0c50

verified ·

1 Parent(s): 2c7d3c3

Update app/services/gemini_service.py

Files changed (1) hide show

app/services/gemini_service.py CHANGED Viewed

@@ -17,7 +17,7 @@ class GeminiNutritionService:
         if not api_key:
             raise ValueError("GEMINI_API_KEY not found in .env")
         if model_name is None:
-            model_name=os.getenv('GEMINI_MODEL','gemini-3-flash-preview')
         self.client=genai.Client(api_key=api_key)
         self.model_name=model_name
     def create_prompt(self,yolo_detections:List[str])->str:

         if not api_key:
             raise ValueError("GEMINI_API_KEY not found in .env")
         if model_name is None:
+            model_name=os.getenv('GEMINI_MODEL','gemini-3.1-flash-lite')
         self.client=genai.Client(api_key=api_key)
         self.model_name=model_name
     def create_prompt(self,yolo_detections:List[str])->str: