Update app.py

app.py

@@ -1,10 +1,9 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load model and tokenizer
-model_name = "models/meta-llama/Meta-Llama-3-8B"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+# Load model and tokenizer directly
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B")
 
 # Define function to translate code
 def translate_code(input_code, prompt=""):
@@ -26,4 +25,4 @@ gr.Interface(
     title="AI Code Translator",
     description="Translate your code using Meta-Llama-3-8B model.",
     theme="compact"
-).launch()
+).launch()