Update app.py

app.py

@@ -12,3 +12,22 @@ def generate_code(prompt,max_length=100,temperature=0.7,top_p=0.95):
 
     generated_code=tokenizer.decode(outputs[0],skip_special_tokens=True)
     return generated_code
+
+#gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## CODE GENERATION WITH CODEGEN MODEL")
+
+    #input bix to add prompt
+    prompt=gr.Textbox(lines=10, label="Enter your prompt for codegen")
+    max_length=gr.Slider(50,500,value=100,label="Max Length")
+    temperature=gr.Slider(0.1,0.9,value=0.7,label="Choose tempertaure")
+    top_p=gr.Slider(0.1,1.0,value=0.95,label="top P value")
+
+    output=gr.Textbox(lines=20, lebel="generated code")
+
+    geberate_button=gr.Button("Generate Code")
+    generage_button.click(fn=generate_code,
+                         inputs=[prompt,max_length,temperature,top_p],
+                         outputs=output)
+    
+demo.launch()