Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import gradio as gr
|
|
| 2 |
import os
|
| 3 |
from transformers import pipeline
|
| 4 |
title = "📗Health and Mindful Story Gen❤️"
|
|
|
|
| 5 |
examples = [
|
| 6 |
["Mental Body Scan"],
|
| 7 |
["Stretch, Calm, Breath"],
|
|
@@ -43,8 +44,12 @@ from gradio.inputs import Textbox
|
|
| 43 |
from gradio import outputs
|
| 44 |
|
| 45 |
HF_TOKEN = os.environ.get("HF_TOKEN") # get token from secrets, copy token value HF_TOKEN from Profile settings token into this repo settings
|
|
|
|
|
|
|
| 46 |
generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN) # add api_key=HF_TOKEN to get over the quota error
|
| 47 |
generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
|
| 48 |
generator1 = gr.Interface.load("huggingface/gpt2-large", api_key=HF_TOKEN)
|
|
|
|
|
|
|
| 49 |
gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
|
| 50 |
title=title, examples=examples).launch(share=False)
|
|
|
|
| 2 |
import os
|
| 3 |
from transformers import pipeline
|
| 4 |
title = "📗Health and Mindful Story Gen❤️"
|
| 5 |
+
|
| 6 |
examples = [
|
| 7 |
["Mental Body Scan"],
|
| 8 |
["Stretch, Calm, Breath"],
|
|
|
|
| 44 |
from gradio import outputs
|
| 45 |
|
| 46 |
HF_TOKEN = os.environ.get("HF_TOKEN") # get token from secrets, copy token value HF_TOKEN from Profile settings token into this repo settings
|
| 47 |
+
|
| 48 |
+
|
| 49 |
generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN) # add api_key=HF_TOKEN to get over the quota error
|
| 50 |
generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)
|
| 51 |
generator1 = gr.Interface.load("huggingface/gpt2-large", api_key=HF_TOKEN)
|
| 52 |
+
|
| 53 |
+
|
| 54 |
gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),
|
| 55 |
title=title, examples=examples).launch(share=False)
|