lucadillenburg commited on
Commit
5282624
·
1 Parent(s): 5d286f4
Files changed (2) hide show
  1. app.py +7 -3
  2. axolotl/src/axolotl/common/cli.py +2 -0
app.py CHANGED
@@ -3,12 +3,16 @@ Chat Interface App
3
  """
4
 
5
  import gradio as gr
6
- from axolotl.cli import print_axolotl_text_art
7
 
8
- print_axolotl_text_art()
 
 
 
 
 
9
 
10
  def generate(instruction):
11
- return "Hello " + instruction + "!"
12
 
13
  iface = gr.Interface(fn=generate, inputs="text", outputs="text")
14
  iface.launch()
 
3
  """
4
 
5
  import gradio as gr
 
6
 
7
+ # from axolotl.cli import print_axolotl_text_art
8
+ # print_axolotl_text_art()
9
+
10
+ import sys
11
+ sys.path.append('axolotl/src/axolotl/common/cli.py')
12
+ import test_func
13
 
14
  def generate(instruction):
15
+ return "Hello " + instruction + "!" + test_func()
16
 
17
  iface = gr.Interface(fn=generate, inputs="text", outputs="text")
18
  iface.launch()
axolotl/src/axolotl/common/cli.py CHANGED
@@ -13,6 +13,8 @@ from axolotl.utils.models import load_model, load_tokenizer
13
  configure_logging()
14
  LOG = logging.getLogger("axolotl.common.cli")
15
 
 
 
16
 
17
  @dataclass
18
  class TrainerCliArgs:
 
13
  configure_logging()
14
  LOG = logging.getLogger("axolotl.common.cli")
15
 
16
+ def test_func():
17
+ return "Hello World to All!"
18
 
19
  @dataclass
20
  class TrainerCliArgs: