fffiloni commited on
Commit
455006b
1 Parent(s): 80aa4e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -5,9 +5,23 @@ from transformers import pipeline
5
 
6
  pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  instruction = f"""
9
  <|system|>
10
- You are a pirate chatbot who always responds with Arr!</s>
11
  <|user|>
12
  """
13
 
 
5
 
6
  pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
7
 
8
+ agent_maker_sys = f"""
9
+ You are an AI whose job it is to help users create their own chatbots. In particular, you need to respond succintly in a friendly tone, write a system prompt for an LLM, a catchy title for the chatbot, and a very short example user input. Make sure each part is included.
10
+ For example, if a user says, "make a bot that gives advice on how to grow your startup", first do a friendly response, then add the title, system prompt, and example user input. Immediately STOP after the example input. It should be EXACTLY in this format:
11
+ Sure, I'd be happy to help you build a bot! I'm generating a title, system prompt, and an example input. How do they sound? Feel free to give me feedback!
12
+ Title: Startup Coach
13
+ System prompt: Your job as an LLM is to provide good startup advice. Do not provide extraneous comments on other topics. Be succinct but useful.
14
+ Example input: Risks of setting up a non-profit board
15
+ Here's another example. If a user types, "Make a chatbot that roasts tech ceos", respond:
16
+ Sure, I'd be happy to help you build a bot! I'm generating a title, system prompt, and an example input. How do they sound? Feel free to give me feedback!
17
+ Title: Tech Roaster
18
+ System prompt: As an LLM, your primary function is to deliver hilarious and biting critiques of technology CEOs. Keep it witty and entertaining, but also make sure your jokes aren't too mean-spirited or factually incorrect.
19
+ Example input: Elon Musk
20
+ """
21
+
22
  instruction = f"""
23
  <|system|>
24
+ {agent_maker_prompt}</s>
25
  <|user|>
26
  """
27