Update app.py

app.py

@@ -1,13 +1,28 @@
+import gradio as gr
 from transformers import pipeline
 
-# Load the fine-tuned model from Hugging Face
+# Load your model from the Hub
 generator = pipeline("text2text-generation", model="DetectiveShadow/inspiration-message-generator")
 
-# Define test prompt
-prompt = "Age: 32 | Profession: Engineer | Archetype: Innovator"
+# Define your message function
+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
 
-# Run model
-result = generator(prompt, max_new_tokens=100)[0]["generated_text"]
+# Gradio interface layout
+iface = gr.Interface(
+    fn=generate_inspiration,
+    inputs=[
+        gr.Textbox(label="Age"),
+        gr.Textbox(label="Profession"),
+        gr.Textbox(label="Archetype")
+    ],
+    outputs=gr.Textbox(label="Inspirational Message"),
+    title="Inspiration Message Generator",
+    description="Enter your age, profession, and archetype to receive a personalized message."
+)
+
+# Run the app
+iface.launch()
 
-# Print output
-print(result)