Update app.py

app.py

@@ -4,14 +4,15 @@ from transformers import pipeline
 
 title = "📗Health and Mindful Story Gen❤️"
 
+# Define the examples to show in the interface
 examples = [
     ["Mental Body Scan"],
     ["Stretch, Calm, Breath"],
     ["Relaxed Seat Breath"],
     ["Walk Feel"],
     ["Brain gamification"],
-    ["alleviating stress"],
-    ["helping breathing, satisfaction"],
+    ["Alleviating stress"],
+    ["Helping breathing, satisfaction"],
     ["Relieve Stress, Build Support"],
     ["Relaxation Response"],
     ["Deep Breaths"],
@@ -41,11 +42,10 @@ examples = [
     ["Learn to reduce feelings of overwhelmed"]
 ]
 
-HF_TOKEN = os.environ.get("HF_TOKEN")  # get token from secrets, copy token value HF_TOKEN from Profile settings token into this repo settings
-
-generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN)
-generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
-generator1 = gr.Interface.load("huggingface/gpt2-large", api_key=HF_TOKEN)
+# Initialize the models using Hugging Face pipelines
+generator1 = pipeline("text-generation", model="gpt2-large")
+generator2 = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+generator3 = pipeline("text-generation", model="EleutherAI/gpt-j-6B")
 
 with gr.Blocks() as demo:
     gr.Markdown(f"# {title}")
@@ -57,9 +57,9 @@ with gr.Blocks() as demo:
         gen3_output = gr.Textbox(label="GPT-J Output")
 
     def generate_outputs(input_text):
-        out1 = generator1([input_text])[0]
-        out2 = generator2([input_text])[0]
-        out3 = generator3([input_text])[0]
+        out1 = generator1(input_text, max_length=50)[0]['generated_text']
+        out2 = generator2(input_text, max_length=50)[0]['generated_text']
+        out3 = generator3(input_text, max_length=50)[0]['generated_text']
         return out1, out2, out3
 
     gr.Button("Generate").click(
@@ -70,4 +70,4 @@ with gr.Blocks() as demo:
 
     gr.Examples(examples=examples, inputs=input_textbox)
 
-demo.launch(share=False)
+demo.launch(share=False)