Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| # Load AI Model | |
| code_generator = pipeline("text-generation", model="Salesforce/codegen-6B-mono") | |
| # Function to generate code from prompts | |
| def generate_code(prompt): | |
| result = code_generator(prompt, max_length=200) | |
| return result[0]['generated_text'] | |
| # Gradio Interface | |
| iface = gr.Interface( | |
| fn=generate_code, | |
| inputs="text", | |
| outputs="text", | |
| title="AI Project Creator", | |
| description="Enter a prompt to generate a full-stack project scaffold." | |
| ) | |
| iface.launch() | |