Create main.py

main.py

@@ -0,0 +1,14 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the Hugging Face model
+model = pipeline("text-generation", model="distilgpt2")
+
+def generate_text(prompt):
+    generated_text = model(prompt, max_length=50)[0]['generated_text']
+    return generated_text
+
+textbox = gr.inputs.Textbox(lines=7, label="Enter a prompt")
+output_text = gr.outputs.Textbox(label="Generated Text")
+
+gr.Interface(fn=generate_text, inputs=textbox, outputs=output_text, title="Hugging Face Text Generation", description="Generate text using a Hugging Face model").launch()