aquibali01/Python-to-javascript-syntax

1import gradio as gr2from transformers import AutoTokenizer, AutoModelForCausalLM3 4# Load the model and tokenizer5model_name = "Salesforce/codegen-350M-mono"6tokenizer = AutoTokenizer.from_pretrained(model_name)7model = AutoModelForCausalLM.from_pretrained(model_name)8 9# Function to translate Python code to JavaScript10def translate_code(python_code):11    inputs = tokenizer(python_code, return_tensors="pt")12    outputs = model.generate(inputs["input_ids"], max_length=150, num_return_sequences=1)13    translated_code = tokenizer.decode(outputs[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)14    return translated_code15 16# Gradio interface17interface = gr.Interface(18    fn=translate_code, 19    inputs="text", 20    outputs="text",21    title="Python to JavaScript Code Translator",22    description="Paste your Python code, and it will be translated into JavaScript.",23)24 25# Launch the app26interface.launch()27