Dagfinn1962 commited on
Commit
01004fd
1 Parent(s): 90387d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -6,7 +6,18 @@ import gradio as gr
6
  from gradio.themes.base import Base
7
  from gradio.themes.utils import colors, fonts, sizes
8
 
9
- def __init__(
 
 
 
 
 
 
 
 
 
 
 
10
  self, system: str = None, user: str = None, assistant: str = None
11
  ) -> None:
12
  if system is not None:
@@ -105,7 +116,7 @@ def call_inf_server(prompt, openai_key):
105
 
106
 
107
  with gr.Blocks(
108
- theme=gr.themes.Glass(),
109
  css=".disclaimer {font-variant-caps: all-small-caps;}",
110
  ) as demo:
111
  gr.Markdown(
 
6
  from gradio.themes.base import Base
7
  from gradio.themes.utils import colors, fonts, sizes
8
 
9
+ from src.llm_boilers import llm_boiler
10
+
11
+
12
+ logging.basicConfig(format="%(asctime)s - %(message)s", level=logging.INFO)
13
+ logging.warning("READY. App started...")
14
+
15
+
16
+ class Chat:
17
+ default_system_prompt = "A conversation between a user and an LLM-based AI assistant. The assistant gives helpful and honest answers."
18
+ system_format = "<|im_start|>system\n{}<|im_end|>\n"
19
+
20
+ def __init__(
21
  self, system: str = None, user: str = None, assistant: str = None
22
  ) -> None:
23
  if system is not None:
 
116
 
117
 
118
  with gr.Blocks(
119
+ theme=gr.themes.Soft(),
120
  css=".disclaimer {font-variant-caps: all-small-caps;}",
121
  ) as demo:
122
  gr.Markdown(