Update vlm/vlm.py

vlm/vlm.py

@@ -2,12 +2,8 @@ import os
 from typing import Any
 from vlm.vlm_interface import VLMIterface
 import google.generativeai as genai
-from dotenv import load_dotenv
-
-# Load environment variables from a .env file
-load_dotenv()
 # Fetch Google API key from environment variables
-_api = os.getenv("GOOGLE_API_KEY")
+API = os.getenv("GOOGLE_API_KEY")
 NAME = "gemini-1.5-flash"
 
 class VLM(VLMIterface):
@@ -22,7 +18,7 @@ class VLM(VLMIterface):
         """
         super().__init__()
         # Configure the generative AI model with the provided API key
-        genai.configure(api_key=_api)
+        genai.configure(api_key=API)
         # Initialize the model (Gemini 1.5 Flash) from Google's generative AI
         self.vlm = genai.GenerativeModel(NAME)