Update app.py

app.py

@@ -17,16 +17,9 @@ model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-rw-1b", trust_remote
 # Function to generate blog content
 def generate_blog(topic, keywords):
     prompt_template = f"""
-    You are a technical content writer. Write a detailed and informative blog on the following topic.
+    You are a content writer. Write a poem of a maximum of 10 sentences on the following topic.
     Topic: {topic}
-    Keywords: {keywords}
-    Make sure the blog covers the following sections:
-    1. Introduction
-    2. Detailed Explanation
-    3. Examples
-    4. Conclusion
-    
-    Blog:
+    Poem:
     """
     input_ids = tokenizer(prompt_template, return_tensors="pt", max_length=512, truncation=True)
     outputs = model.generate(input_ids["input_ids"], max_length=800, num_return_sequences=1)
@@ -38,12 +31,11 @@ def generate_blog(topic, keywords):
 iface = gr.Interface(
     fn=generate_blog,
     inputs=[
-        gr.Textbox(lines=2, placeholder="Enter the blog topic", label="Blog Topic"),
-        gr.Textbox(lines=2, placeholder="Enter keywords (comma-separated)", label="Keywords")
+        gr.Textbox(lines=2, placeholder="Enter the poem topic", label="Blog Topic"),
     ],
-    outputs=gr.Textbox(label="Generated Blog Content"),
-    title="Technical Blog Generator",
-    description="Generate a detailed technical blog by providing a topic and relevant keywords."
+    outputs=gr.Textbox(label="Generated Poem"),
+    title="Poem Generator",
+    description="Generate a poem based on the providing a topic."
 )
 
 if __name__ == "__main__":