Create app.py

The Gradio App

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# Load the small model (WizardCoder-1B)
+model_name = "WizardLM/WizardCoder-1B-V1.0"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")
+
+def generate_manim_code(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
+    outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.7)
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return response
+
+description = "💡 Enter a prompt like: *Write a Manim script to animate a growing square*"
+
+iface = gr.Interface(
+    fn=generate_manim_code,
+    inputs=gr.Textbox(label="Prompt", placeholder="e.g., Create a bouncing ball using Manim"),
+    outputs=gr.Textbox(label="Generated Manim Code"),
+    title="🎬 Manim Code Generator",
+    description=description
+)
+
+iface.launch()