Update app.py

app.py

@@ -1,19 +1,22 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM
+from transformers import pipeline
 
-# Load the Llama-2-7b-chat model
-model = AutoModelForCausalLM.from_pretrained("llama-2-7b-chat")
+generator = pipeline('text-generation', model='llama-2-7b-chat')
 
-# Define a function to generate text using the model
-def generate_text(prompt):
-  # Generate text using the model
-  generated_text = model.generate(prompt, max_length=4000)
+def generate(text):
+    result = generator(text, max_length=100, num_return_sequences=1)
+    return result[0]["generated_text"]
 
-  # Return the generated text
-  return generated_text
+examples = [
+    ["The Moon's orbit around Earth has"],
+    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
+]
 
-# Create a Gradio app
-interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    examples=examples
+)
 
-# Launch the Gradio app
-interface.launch()
+demo.launch()