Update app.py

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the model directly from Hugging Face Model Hub
+generator = pipeline("text-generation", model="LiquidAI/LFM2-1.2B")
+
+# Function to generate text from input prompt
+def generate_text(prompt):
+    result = generator(prompt, max_length=50, num_return_sequences=1)
+    return result[0]['generated_text']
+
+# Create a Gradio interface
+iface = gr.Interface(fn=generate_text,
+                     inputs="text",
+                     outputs="text",
+                     title="GPT-2 Text Generation",
+                     description="Enter a prompt and see the text generated by GPT-2!")
+
+# Launch the interface
+iface.launch()