Update app.py

app.py

@@ -1,10 +1,63 @@
 import gradio as gr
 import spaces
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-@spaces.GPU
-def echo(x: str) -> str:
-    # A trivial GPU-decorated function to satisfy ZeroGPU requirement
-    return f"✅ Works! You said: {x}"
+# -------------------------------
+# Load Qwen2.5-7B for text-only chat
+# -------------------------------
+model_id = "Qwen/Qwen2.5-7B-Instruct"
+
+tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    torch_dtype=torch.float16,     # use half precision for GPU memory
+    device_map="auto",             # map layers automatically to GPU
+    trust_remote_code=True
+)
+
+qwen_pipe = pipeline(
+    "text-generation",
+    model=model,
+    tokenizer=tokenizer,
+    device_map="auto"
+)
+
+# -------------------------------
+# Life Genie function
+# -------------------------------
+@spaces.GPU  # ensure HF detects GPU usage
+def life_genie(user_input: str) -> str:
+    """
+    Life Genie: magical assistant powered by Qwen2.5-7B.
+    """
+    if not user_input.strip():
+        return "🧞 Please ask me something magical!"
+    
+    prompt = f"You are a witty magical genie. Answer creatively.\nUser: {user_input}\nGenie:"
+    output = qwen_pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7)
+    return output[0]["generated_text"]
+
+# -------------------------------
+# Gradio UI
+# -------------------------------
+with gr.Blocks(css="""
+body {
+    background: linear-gradient(135deg, #ff9966, #ff5e62);
+    font-family: Trebuchet MS, sans-serif;
+}
+""") as demo:
+
+    gr.HTML("<h1 style='text-align:center; color:white;'>🧞✨ Life Genie — Ask Your Magical Assistant ✨🧞</h1>")
+    gr.Markdown("Type your wish below and summon the genie for a response!")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            user_box = gr.Textbox(lines=3, placeholder="E.g. Cheer me up about Mondays 🪄", label="Your Wish")
+            submit_btn = gr.Button("✨ Summon Genie")
+        with gr.Column(scale=1):
+            genie_out = gr.Textbox(lines=10, label="Genie Responds")
+
+    submit_btn.click(life_genie, inputs=user_box, outputs=genie_out)
 
-demo = gr.Interface(fn=echo, inputs="text", outputs="text")
 demo.launch()