Update app.py

app.py

@@ -1,16 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load your model from the Hub
-generator = pipeline("text2text-generation", model="DetectiveShadow/inspiration-message-generator")
+# Load your fine-tuned model from Hugging Face Hub
+generator = pipeline(
+    "text2text-generation",
+    model="DetectiveShadow/inspiration-message-generator"  # ← your model name
+)
 
-# Define your message function
+# Function to create the input prompt and get the message
 def generate_inspiration(age, profession, archetype):
     prompt = f"Age: {age} | Profession: {profession} | Archetype: {archetype}"
-    result = generator(prompt, max_new_tokens=100)[0]["generated_text"]
-    return result
+    result = generator(prompt, max_new_tokens=100)
+    return result[0]["generated_text"]
 
-# Gradio interface layout
+# Gradio interface
 iface = gr.Interface(
     fn=generate_inspiration,
     inputs=[
@@ -20,9 +23,8 @@ iface = gr.Interface(
     ],
     outputs=gr.Textbox(label="Inspirational Message"),
     title="Inspiration Message Generator",
-    description="Enter your age, profession, and archetype to receive a personalized message."
+    description="Get a personalized motivational message based on your age, profession, and life archetype."
 )
 
-# Run the app
 iface.launch()