This project is an AI-powered Code Generation Tool built using CodeGen model, integrated into a simple web UI using Gradio. You can input prompts like "write a bubble sort function" and the AI will generate Python code instantly.
- Code generation from text prompts
- Runs on Google Colab (with Tesla T4 GPU support)
- Lightweight, fast inference
- Easy-to-use web interface with Gradio
- Model: CodeGen (Salesforce/codegen-350M-multi)
app.py
β Main app with Gradio interfacemodel_loader.py
β Loads the model and tokenizerinterface.py
β Frontend logicrequirements.txt
β Dependencies to run locally (optional)
- Open
app.py
in Colab - Run all cells
- The Gradio interface will appear to generate code!
- Support more models (Code LLaMA, StarCoder, etc.)
- Add download/save option for generated code
- Improve prompt understanding