Update app.py

app.py

@@ -19,39 +19,30 @@ FIM_SUFFIX = "<fim_suffix>"
 FIM_INDICATOR = "<FILL_HERE>"
 
 FORMATS = """## Model Formats
-
 The model is pretrained on code and is formatted with special tokens in addition to the pure code data,\
 such as prefixes specifying the source of the file or tokens separating code from a commit message.\
 Use these templates to explore the model's capacities:
-
 ### 1. Prefixes 🏷️
 For pure code files, use any combination of the following prefixes:
-
 ```
 <reponame>REPONAME<filename>FILENAME<gh_stars>STARS\ncode<|endoftext|>
 ```
 STARS can be one of: 0, 1-10, 10-100, 100-1000, 1000+
-
 ### 2. Commits 💾
 The commits data is formatted as follows:
-
 ```
 <commit_before>code<commit_msg>text<commit_after>code<|endoftext|>
 ```
-
 ### 3. Jupyter Notebooks 📓
 The model is trained on Jupyter notebooks as Python scripts and structured formats like:
-
 ```
 <start_jupyter><jupyter_text>text<jupyter_code>code<jupyter_output>output<jupyter_text>
 ```
-
 ### 4. Issues 🐛
 We also trained on GitHub issues using the following formatting:
 ```
 <issue_start><issue_comment>text<issue_comment>...<issue_closed>
 ```
-
 ### 5. Fill-in-the-middle 🧩
 Fill in the middle requires rearranging the model inputs. The playground handles this for you - all you need is to specify where to fill:
 ```
@@ -84,31 +75,40 @@ clients = {
         {"Wiki": tool_fn},
         """\
 Answer the following question:
-
 Q: In which branch of the arts is Patricia Neary famous?
 A: Ballets
 A2: <request><Wiki>Patricia Neary<call>Patricia Neary (born October 27, 1942) is an American ballerina, choreographer and ballet director, who has been particularly active in Switzerland. She has also been a highly successful ambassador for the Balanchine Trust, bringing George Balanchine's ballets to 60 cities around the globe.<response>
 Result=Ballets<submit>
-
 Q: Who won Super Bowl XX?
 A: Chicago Bears
 A2: <request><Wiki>Super Bowl XX<call>Super Bowl XX was an American football game between the National Football Conference (NFC) champion Chicago Bears and the American Football Conference (AFC) champion New England Patriots to decide the National Football League (NFL) champion for the 1985 season. The Bears defeated the Patriots by the score of 46–10, capturing their first NFL championship (and Chicago's first overall sports victory) since 1963, three years prior to the birth of the Super Bowl. Super Bowl XX was played on January 26, 1986 at the Louisiana Superdome in New Orleans.<response>
 Result=Chicago Bears<submit>
 """
     ],
-    "StarCoderBase TriviaQA2": [
+    "StarCoderBase GSM8K": [
         Client(
-            "https://api-inference.huggingface.co/models/vwxyzjn/starcoderbase-triviaqa",
+            "https://api-inference.huggingface.co/models/lvwerra/starcoderbase-gsm8k",
             headers={"Authorization": f"Bearer {HF_TOKEN}"},
         ),
-        {"Wiki": tool_fn},
+        {"Wiki": load_tool("lvwerra/python-interpreter")},
         """\
-Answer the following question:
-
-Q: In which branch of the arts is Patricia Neary famous?
-A: Ballets
-A2: <request><Wiki>Patricia Neary<call>Patricia Neary (born October 27, 1942) is an American ballerina, choreographer and ballet director, who has been particularly active in Switzerland. She has also been a highly successful ambassador for the Balanchine Trust, bringing George Balanchine's ballets to 60 cities around the globe.<response>
-Result=Ballets<submit>
+Example of using a Python API to solve math questions. 
+
+Q: Olivia has $23. She bought five bagels for $3 each. How much money does she have left?
+
+<request><PythonInterpreter>
+def solution():
+    money_initial = 23
+    bagels = 5
+    bagel_cost = 3
+    money_spent = bagels * bagel_cost
+    money_left = money_initial - money_spent
+    result = money_left
+    return result
+print(solution())
+<call>72<response>
+
+Result = 72 <submit>
 """
     ],
 }
@@ -393,15 +393,12 @@ demo.queue(concurrency_count=16).launch(debug=True)
 
 """
 Answer the following question:
-
 Q: In which branch of the arts is Patricia Neary famous?
 A: Ballets
 A2: <request><Wiki>Patricia Neary<call>Patricia Neary (born October 27, 1942) is an American ballerina, choreographer and ballet director, who has been particularly active in Switzerland. She has also been a highly successful ambassador for the Balanchine Trust, bringing George Balanchine's ballets to 60 cities around the globe.<response>
 Result=Ballets<submit>
-
 Q: Who won Super Bowl XX?
 A: Chicago Bears
 A2: <request><Wiki>Super Bowl XX<call>Super Bowl XX was an American football game between the National Football Conference (NFC) champion Chicago Bears and the American Football Conference (AFC) champion New England Patriots to decide the National Football League (NFL) champion for the 1985 season. The Bears defeated the Patriots by the score of 46–10, capturing their first NFL championship (and Chicago's first overall sports victory) since 1963, three years prior to the birth of the Super Bowl. Super Bowl XX was played on January 26, 1986 at the Louisiana Superdome in New Orleans.<response>
 Result=Chicago Bears<submit>
-
 Q: In what state is Philadelphia located?"""