Update app.py

app.py

@@ -12,7 +12,7 @@ sequence_length = 150
 
 # Load API keys
 HF_API_TOKEN = os.getenv("HF_API_TOKEN")
-GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
+# GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
 
 # Hugging Face setup
 MODEL_ID = "Salesforce/codet5p-770m"
@@ -20,7 +20,7 @@ API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
 # Gemini setup
-genai.configure(api_key="AIzaSyBkc8CSEhyYwZAuUiJfzF1Xtns-RYmBOpg")
+genai.configure(api_key="AIzaSyBlU-T3bx0A_ztKqqDFayOEqdjgQHifsf4")
 
 # --- Load Local Model & Vectorizers ---
 model = tf.keras.models.load_model("java_to_python_seq2seq_model.h5")
@@ -46,7 +46,7 @@ def fallback_translate_with_gemini(code_snippet, source_lang, target_lang):
     Do not give any explanation. Only give the translated code.
     """
     try:
-        model = genai.GenerativeModel("gemini-1.5-pro")
+        model = genai.GenerativeModel("gemini-2.0-flash-lite")
         response = model.generate_content(prompt)
         return response.text.strip() if response else "Translation failed."
     except Exception as e: