Update main.py

main.py

@@ -5,9 +5,14 @@ import google.generativeai as genai
 from dotenv import load_dotenv
 import io
 from PIL import Image
+import os
 
 app = FastAPI()
 
+API_KEY=os.getenv('GEMINI_API_KEY')
+
+genai.configure(api_key=API_KEY)
+
 def get_gemini_response(image: bytes) -> Any:
     image_stream = io.BytesIO(image)
     image = Image.open(image_stream)