Update app.py

app.py

@@ -2,17 +2,14 @@
 RWKV RNN Model - Gradio Space for HuggingFace
 YT - Mean Gene Hacks - https://www.youtube.com/@MeanGeneHacks
 (C) Gene Ruebsamen - 2/7/2023
-
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
-
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
-
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 """
@@ -85,7 +82,13 @@ def infer(
     # Clear model state for generative mode
     model.resetState()
     if (mode == "Q/A"):
-        prompt = f"Ask Expert\n\nQuestion:\n{prompt}\n\nExpert Full Answer:\n"
+        prompt = f"\nQ: {prompt}\n\nA:"
+    if (mode == "ELDR"):
+        prompt = f"\n{prompt}\n\nExpert Long Detailed Response:"
+    if (mode == "EFA"):
+        prompt = f'\nAsk Expert\n\nQuestion:\n{prompt}\n\nExpert Full Answer:\n'
+    if (mode == "BFR"):
+        prompt = f"Task given:\n\n{prompt}\n\nBest Full Response:"
 
     print(f"PROMPT ({datetime.now()}):\n-------\n{prompt}")
     print(f"OUTPUT ({datetime.now()}):\n-------\n")
@@ -146,7 +149,6 @@ def chat(
     username = username or "USER"
 
     intro = f'''The following is a verbose and detailed conversation between an AI assistant called FRITZ, and a human user called USER. FRITZ is intelligent, knowledgeable, wise and polite.
-
     {username}: What year was the french revolution?
     FRITZ: The French Revolution started in 1789, and lasted 10 years until 1799.
     {username}: 3+5=?
@@ -218,7 +220,6 @@ examples = [
     [
         # Chatbot
         '''This is a conversation between two AI large language models named Alex and Fritz. They are exploring each other's capabilities, and trying to ask interesting questions of one another to explore the limits of each others AI.
-
 Conversation:
 Alex: Good morning, Fritz, what type of LLM are you based upon?
 Fritz: Morning Alex, I am an RNN with transformer level performance. My language model is 100% attention free.
@@ -226,9 +227,7 @@ Alex:''', "generative", 220, 0.9, 0.9, "\\n\\n,<|endoftext|>"],
     [
         # Generate List
         '''Task given:
-
 Please Write a Short story about a cat learning python
-
 Best Full Response:
 ''', "generative", 140, 0.85, 0.8, "<|endoftext|>"],
     [
@@ -244,7 +243,7 @@ iface = gr.Interface(
     allow_flagging="never",
     inputs=[
         gr.Textbox(lines=20, label="Prompt"),  # prompt
-        gr.Radio(["generative", "Q/A"],
+        gr.Radio(["generative", "Q/A","ELDR","EFR","BFR"],
                  value="generative", label="Choose Mode"),
         gr.Slider(1, 256, value=40),  # max_tokens
         gr.Slider(0.0, 1.0, value=0.8),  # temperature
@@ -282,4 +281,4 @@ demo = gr.TabbedInterface(
 )
 
 demo.queue()
-demo.launch(share=False)
+demo.launch(share=False)