awacke1 commited on
Commit
d07b728
·
1 Parent(s): 537f363

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -4,12 +4,12 @@ import requests
4
  # GPT-J-6B API
5
  API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
6
  headers = {"Authorization": "Bearer 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"}
7
- prompt = """Carl Marx: Blah blah democracy blah blah
8
- Biden: Hey man will you shut up man?
9
- Einstein: Science I tell you
10
- Elon Musk: Give me some science man!"""
11
 
12
- examples = [["river"], ["night"], ["trees"],["table"],["laughs"]]
13
 
14
 
15
  def poem2_generate(word):
 
4
  # GPT-J-6B API
5
  API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
6
  headers = {"Authorization": "Bearer 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"}
7
+ prompt = """Customer: Hi I would like to get some help with my stress and anxiety
8
+ Agent: I should be able to help you. I have some Wellness and Mindfulness advice.
9
+ Customer: Great! I've been searching for good solutions to enhance memory and health
10
+ Agent: Let me share some of the resources with you. We will have you feeling great in no time at all with tips like nutrition, exercise, and good choices"""
11
 
12
+ examples = [["mind"], ["memory"], ["sleep"],["wellness"],["nutrition"]]
13
 
14
 
15
  def poem2_generate(word):