Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ sequence_length = 150
|
|
| 12 |
|
| 13 |
# Load API keys
|
| 14 |
HF_API_TOKEN = os.getenv("HF_API_TOKEN")
|
| 15 |
-
GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 16 |
|
| 17 |
# Hugging Face setup
|
| 18 |
MODEL_ID = "Salesforce/codet5p-770m"
|
|
@@ -45,7 +45,7 @@ def fallback_translate_with_gemini(code_snippet, source_lang, target_lang):
|
|
| 45 |
Do not give any explanation. Only give the translated code.
|
| 46 |
"""
|
| 47 |
try:
|
| 48 |
-
model = genai.GenerativeModel("gemini-
|
| 49 |
response = model.generate_content(prompt)
|
| 50 |
return response.text.strip() if response else "Translation failed."
|
| 51 |
except Exception as e:
|
|
|
|
| 12 |
|
| 13 |
# Load API keys
|
| 14 |
HF_API_TOKEN = os.getenv("HF_API_TOKEN")
|
| 15 |
+
# GEMINI_API_KEY = os.getenv("GOOGLE_API_KEY")
|
| 16 |
|
| 17 |
# Hugging Face setup
|
| 18 |
MODEL_ID = "Salesforce/codet5p-770m"
|
|
|
|
| 45 |
Do not give any explanation. Only give the translated code.
|
| 46 |
"""
|
| 47 |
try:
|
| 48 |
+
model = genai.GenerativeModel("gemini-2.0-flash-lite")
|
| 49 |
response = model.generate_content(prompt)
|
| 50 |
return response.text.strip() if response else "Translation failed."
|
| 51 |
except Exception as e:
|