Update app.py
Browse filesadded more API keys
app.py
CHANGED
|
@@ -9,6 +9,7 @@ demo = gr.Blocks()
|
|
| 9 |
|
| 10 |
name_list = ['huggingface/bigscience/T0pp', 'huggingface/EleutherAI/gpt-j-6B', 'huggingface/gpt2-xl', 'huggingface/EleutherAI/gpt-neo-2.7B']
|
| 11 |
|
|
|
|
| 12 |
examples = [#zero-shot
|
| 13 |
["Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA: The answer (arabic numerals) is "],
|
| 14 |
#zero-shot-CoT
|
|
@@ -18,7 +19,6 @@ examples = [#zero-shot
|
|
| 18 |
#few-shot-CoT
|
| 19 |
["Q: Roger has 5 tennis balls. He buys 2 more cans of tennis balls. Each can has 3 tennis balls. How many tennis balls does he have now?\nA: Roger started with 5 balls. 2 cans of 3 tennis balls each is 6 tennis balls. 5 + 6 = 11. The answer is 11.\nQ:A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA:"],
|
| 20 |
]
|
| 21 |
-
#examples from Figure 1 of the paper
|
| 22 |
|
| 23 |
def calculator(num1, operation, num2):
|
| 24 |
if operation == "add":
|
|
@@ -30,11 +30,16 @@ def calculator(num1, operation, num2):
|
|
| 30 |
elif operation == "divide":
|
| 31 |
return num1 / num2
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
def complete_with_gpt(text):
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
|
|
|
| 38 |
return [interface(text) for interface in interfaces]
|
| 39 |
|
| 40 |
def set_example(example: list) -> dict:
|
|
|
|
| 9 |
|
| 10 |
name_list = ['huggingface/bigscience/T0pp', 'huggingface/EleutherAI/gpt-j-6B', 'huggingface/gpt2-xl', 'huggingface/EleutherAI/gpt-neo-2.7B']
|
| 11 |
|
| 12 |
+
#examples from Figure 1 of the paper
|
| 13 |
examples = [#zero-shot
|
| 14 |
["Q: A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA: The answer (arabic numerals) is "],
|
| 15 |
#zero-shot-CoT
|
|
|
|
| 19 |
#few-shot-CoT
|
| 20 |
["Q: Roger has 5 tennis balls. He buys 2 more cans of tennis balls. Each can has 3 tennis balls. How many tennis balls does he have now?\nA: Roger started with 5 balls. 2 cans of 3 tennis balls each is 6 tennis balls. 5 + 6 = 11. The answer is 11.\nQ:A juggler can juggle 16 balls. Half of the balls are golf balls, and half of the golf balls are blue. How many blue golf balls are there?\nA:"],
|
| 21 |
]
|
|
|
|
| 22 |
|
| 23 |
def calculator(num1, operation, num2):
|
| 24 |
if operation == "add":
|
|
|
|
| 30 |
elif operation == "divide":
|
| 31 |
return num1 / num2
|
| 32 |
|
| 33 |
+
|
| 34 |
+
secrets = [API_KEY1, API_KEY2, API_KEY3, API_KEY4, API_KEY5, API_KEY6]
|
| 35 |
+
|
| 36 |
+
|
| 37 |
def complete_with_gpt(text):
|
| 38 |
+
for secret in secrets:
|
| 39 |
+
try:
|
| 40 |
+
interfaces = [gr.Interface.load(name, api_key = "secret") for name in name_list]
|
| 41 |
+
except:
|
| 42 |
+
print("Error: API key is not valid")
|
| 43 |
return [interface(text) for interface in interfaces]
|
| 44 |
|
| 45 |
def set_example(example: list) -> dict:
|