Update app.py

app.py

@@ -1,25 +1,13 @@
-import gradio as gr
 from transformers import pipeline
 
-# Load the model from the hub
+# Load the fine-tuned model from Hugging Face
 generator = pipeline("text2text-generation", model="DetectiveShadow/inspiration-message-generator")
 
-def generate_message(age, profession, archetype):
-    prompt = f"Age: {age} | Profession: {profession} | Archetype: {archetype}"
-    result = generator(prompt, max_new_tokens=100)[0]["generated_text"]
-    return result
+# Define test prompt
+prompt = "Age: 32 | Profession: Engineer | Archetype: Innovator"
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_message,
-    inputs=[
-        gr.Textbox(label="Age"),
-        gr.Textbox(label="Profession"),
-        gr.Textbox(label="Archetype")
-    ],
-    outputs="text",
-    title="Inspiration Message Generator",
-    description="Enter age, profession, and archetype to get a personalized inspirational message."
-)
+# Run model
+result = generator(prompt, max_new_tokens=100)[0]["generated_text"]
 
-iface.launch()
+# Print output
+print(result)