Update app.py

app.py

@@ -9,7 +9,7 @@ API_TOKEN = os.getenv("HF_API_TOKEN")
 # Change MODEL_ID to a better model
 # MODEL_ID = "Salesforce/codet5p-770m"  # CodeT5+ (Recommended)
 # MODEL_ID = "bigcode/starcoder2-15b"  # StarCoder2
-MODEL_ID = "Salesforce/codet5p-770m"  # Code Llama
+MODEL_ID = "Salesforce/CodeT5+"  # Code Llama
 # MODEL_ID = "bigcode/starcoder"
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
 HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
@@ -23,8 +23,8 @@ def translate_code(code_snippet, source_lang, target_lang):
         "parameters": {
             "max_new_tokens": 150,
             "temperature": 0.2,
-            "top_k": 50
-            # "stop": ["\n\n", "#", "//", "'''"]
+            "top_k": 50,
+            "stop": ["\n\n", "#", "//", "'''"]
         }
     })