waleed-12/pseudo-code-to-Code

1import streamlit as st2from transformers import AutoTokenizer, AutoModelForCausalLM3import torch4 5# Load model and tokenizer6MODEL_NAME = "Johannes/code-generation-model-fine-tuned"7tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)8model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)9 10st.title("Pseudo-code to Code")11 12prompt = st.text_area("Enter a code:")13 14# Fixed maximum length15max_length = 15016 17if st.button("Generate Code"):18    if prompt.strip() != "":19        inputs = tokenizer(prompt, return_tensors="pt")20        outputs = model.generate(21            **inputs,22            max_length=max_length,23            do_sample=True,24            temperature=0.725        )26        generated_code = tokenizer.decode(outputs[0], skip_special_tokens=True)27        st.code(generated_code, language="python")28    else:29        st.warning("Please enter a prompt.")30