https://github.com/mohitkumar5563/ai-codegen-gradio
AI-Assisted Code Generation with CodeGen and Gradio
https://github.com/mohitkumar5563/ai-codegen-gradio
ai codegen colab gradio gradio-python-llm machine-learning nlp python
Last synced: 7 months ago
JSON representation
AI-Assisted Code Generation with CodeGen and Gradio
- Host: GitHub
- URL: https://github.com/mohitkumar5563/ai-codegen-gradio
- Owner: Mohitkumar5563
- Created: 2025-04-05T08:51:39.000Z (7 months ago)
- Default Branch: main
- Last Pushed: 2025-04-05T09:37:46.000Z (7 months ago)
- Last Synced: 2025-04-09T19:16:11.660Z (7 months ago)
- Topics: ai, codegen, colab, gradio, gradio-python-llm, machine-learning, nlp, python
- Language: Python
- Homepage:
- Size: 4.88 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 🤖 AI Code Generation Project
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.
## 🚀 Features
- 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)
## 📁 Files
- `app.py` – Main app with Gradio interface
- `model_loader.py` – Loads the model and tokenizer
- `interface.py` – Frontend logic
- `requirements.txt` – Dependencies to run locally (optional)
## 🧠 Example Prompt
# 🌐 How to Run (on Colab)
1. Open `app.py` in Colab
2. Run all cells
3. The Gradio interface will appear to generate code!
## 📦 Future Enhancements
- Support more models (Code LLaMA, StarCoder, etc.)
- Add download/save option for generated code
- Improve prompt understanding
----