Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load the model and tokenizer
 model_name = "Salesforce/codegen-350M-mono"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Function to translate Python code to JavaScript
 def translate_code(python_code):
-    inputs = tokenizer(python_code, return_tensors="pt", max_length=512, truncation=True)
-    outputs = model.generate(**inputs, max_length=512, num_beams=5, early_stopping=True)
-    translated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    inputs = tokenizer(python_code, return_tensors="pt")
+    outputs = model.generate(inputs["input_ids"], max_length=150, num_return_sequences=1)
+    translated_code = tokenizer.decode(outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
     return translated_code
 
 # Gradio interface