Update app.py

app.py

@@ -1,4 +1,5 @@
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import gradio as gr
 
 # Load the pre-trained GPT2 model and tokenizer
 model = GPT2LMHeadModel.from_pretrained("gpt2")
@@ -29,7 +30,16 @@ def generate_code(prompt, max_length=200):
     
     return code
 
-# Example usage
-prompt = "a function to calculate the factorial of a number"
-generated_code = generate_code(prompt)
-print(generated_code)
+def predict(input_text):
+    output = generate_code(input_text)
+    return output
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Textbox(lines=2, placeholder="Enter your code generation prompt here..."),
+    outputs="text",
+    title="Code Generation with GPT2",
+    description="Generate Python code based on your input prompt."
+)
+
+iface.launch()