Create app.py

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from transformers import pipeline
+
+# Code Generation Pipeline
+code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono")
+
+def generate_code(prompt):
+    output = code_pipeline(prompt, max_length=50)
+    return output[0]['generated_text']
+
+# Gradio Interface
+demo = gr.Interface(
+    fn=generate_code,
+    inputs=gr.Textbox(label="Enter Code Prompt"),
+    outputs=gr.Textbox(label="Generated Code"),
+    title="Code Generation with Transformers",
+    description="Enter a Python function header or code snippet, and the model will generate the rest."
+)
+
+demo.launch()