app.py

"""
Chat Interface App
"""

import gradio as gr

# from axolotl.cli import print_axolotl_text_art
# print_axolotl_text_art()

import sys
sys.path.append('axolotl/src/axolotl/common/cli.py')
from cli import test_func

def generate(instruction):
    return "Hello " + instruction + "!" + test_func()

iface = gr.Interface(fn=generate, inputs="text", outputs="text")
iface.launch()