Update app.py

app.py

@@ -1,7 +1,22 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Initialize the text generation pipeline
+pipe = pipeline("text-generation", model="meta-llama/CodeLlama-7b-hf")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+def generate_text(prompt):
+    # Generate text based on the provided prompt using the pipeline
+    generated_text = pipe(prompt, max_length=50, do_sample=True)[0]['generated_text']
+    return generated_text
+
+# Create a Gradio interface
+demo = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    title="Text Generation with CodeLlama-7b-hf",
+    description="Enter a prompt and see the generated text."
+)
+
+# Launch the Gradio interface
 demo.launch()