initial app.py

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load the TinyLlama text generation pipeline
+pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+
+# Define the inference function
+def generate_text(prompt):
+    results = pipe(prompt, max_length=50, num_return_sequences=1)
+    return results[0]['generated_text']
+
+# Create a Gradio interface
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="TinyLlama Text Generation")
+
+# Launch the interface
+iface.launch()