Create app.py

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+from transformers import pipeline
+
+# Pre-load multiple models
+models = {
+    "TinyBERT (NLU)": pipeline("fill-mask", model="prajjwal1/bert-tiny"),
+    "DistilBERT (NLU)": pipeline("fill-mask", model="distilbert-base-uncased"),
+    "ALBERT (NLU)": pipeline("fill-mask", model="albert-base-v2"),
+    "MobileBERT (NLU)": pipeline("fill-mask", model="google/mobilebert-uncased"),
+    "GPT-2 (text gen)": pipeline("text-generation", model="gpt2")
+}
+
+def run_model(model_name, text):
+    pipe = models[model_name]
+    if "GPT-2" in model_name:
+        return pipe(text, max_length=50, do_sample=True)[0]["generated_text"]
+    else:
+        return str(pipe(text))
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🔥 Tiny LLM Playground\nChoose a small model and test it!")
+    
+    model_choice = gr.Dropdown(list(models.keys()), label="Choose Model")
+    text_input = gr.Textbox(label="Enter text or prompt")
+    output = gr.Textbox(label="Output")
+    
+    run_btn = gr.Button("Run")
+    run_btn.click(fn=run_model, inputs=[model_choice, text_input], outputs=output)
+
+demo.launch()