Update app_cpu.py

app_cpu.py

@@ -1,18 +1,28 @@
+import os
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from huggingface_hub import snapshot_download
 
 # =========================================================
-# Load model and tokenizer
+# 1️⃣ Download model to local cache once
 # =========================================================
 MODEL_ID = "gokaygokay/prompt-enhancer-gemma-3-270m-it"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
+print(f"📦 Downloading model {MODEL_ID} ...")
+model_path = snapshot_download(repo_id=MODEL_ID)
+print(f"✅ Model downloaded to: {model_path}")
+
+# =========================================================
+# 2️⃣ Load tokenizer and model from local path
+# =========================================================
+tokenizer = AutoTokenizer.from_pretrained(model_path)
+model = AutoModelForCausalLM.from_pretrained(model_path)
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 # =========================================================
-# Define enhancer function
+# 3️⃣ Define function to enhance prompt
 # =========================================================
 def enhance_prompt(prompt: str):
+    """Enhance and expand a user prompt with more details and context."""
     if not prompt.strip():
         return "⚠️ Please enter a prompt to enhance."
 
@@ -27,13 +37,14 @@ def enhance_prompt(prompt: str):
     return result.strip()
 
 # =========================================================
-# Gradio UI
+# 4️⃣ Gradio UI
 # =========================================================
 with gr.Blocks(theme=gr.themes.Soft(), title="Prompt Enhancer ✨") as demo:
     gr.Markdown(
         """
         # ✨ Prompt Enhancer — Gemma 3 270M IT  
-        Enhance and expand your prompts with creative detail and context using a small, efficient language model.
+        Give your idea a creative boost!  
+        Enter a simple prompt, and this app will enhance it with vivid detail and context.
         """
     )
 
@@ -57,7 +68,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Prompt Enhancer ✨") as demo:
     gr.Markdown(
         """
         ---
-        🧠 **Tip:** Try short creative prompts like  
+        🧠 **Try these examples:**  
         - “a futuristic city at sunset”  
         - “a woman reading under a tree”  
         - “a magical forest with glowing mushrooms”
@@ -65,7 +76,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Prompt Enhancer ✨") as demo:
     )
 
 # =========================================================
-# Launch the app
+# 5️⃣ Launch app
 # =========================================================
 if __name__ == "__main__":
     demo.launch()