Added .gitignore

.env

@@ -1 +0,0 @@
-GOOGLE_API_KEY='AIzaSyC5vclIaHLdDDOJ04bmkjmF3VGu3v1wOTQ'

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -2,7 +2,6 @@ from PIL import Image
 from io import BytesIO
 from google import genai
 from google.genai import types
-from dotenv import load_dotenv
 from transforms import transform, target_transform
 from torchvision import models
 from hf_class import MyResNet18
@@ -11,8 +10,6 @@ import torch.nn.functional as F
 import os
 import gradio as gr
 
-load_dotenv()
-
 index_to_label = {
     0: 'battery',
     1: 'biological',
@@ -40,7 +37,9 @@ def predict_image(img: Image):
     predicted_label = index_to_label[preds.item()]
     confidence_score = float('{:.4f}'.format(softmax[0][preds.item()].item()))
 
-    client = genai.Client(api_key=os.getenv('GOOGLE_API_KEY'))
+    google_api_key = os.environ.get('GOOGLE_API_KEY')
+
+    client = genai.Client(api_key=google_api_key)
 
     if predicted_label != 'trash':
         response = client.models.generate_content(

test.py

@@ -0,0 +1,4 @@
+import os
+
+api_key = os.environ.get("GOOGLE_API_KEY")
+print(api_key)