Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import time
 
 device = torch.device('cuda')
 
@@ -14,6 +15,9 @@ tokenizer.to(device)
 
 def generate_code(prompt):
     full_prompt = f"# HTML/CSS/JS website code for:\n# {prompt}\n"
+    if prompt == "OzLabsWereHere":
+        time.sleep(10)
+        return "M'en va t'à la fontaine pour y pêcher du poisson, la Ziguezon zinzon"
     inputs = tokenizer(full_prompt, return_tensors="pt").input_ids
     outputs = model.generate(inputs, max_length=1024, temperature=0.7)
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -24,6 +28,6 @@ gr.Interface(
     fn=generate_code,
     inputs=gr.Textbox(lines=6, label="Describe your website idea"),
     outputs=gr.Code(label="Generated HTML/CSS/JS"),
-    title="SKORD AI Website Generator",
+    title="Oz AI Website Generator",
     description="Describe your idea in English or Hindi. This will return only HTML/CSS/JS."
 ).launch()