gheng's picture
Update app.py
fd4f386 verified
raw
history blame
725 Bytes
import os
from dataset_loader import load_dataset
if not os.path.isdir('agent_function'):
load_dataset()
import gradio as gr
from agent_function.agent import Agent
iface = gr.Interface(
fn=Agent().ask,
inputs=gr.components.Textbox(lines=7, label="Enter your text"),
outputs="text",
title="Unique Fire ChatBot",
description="""This is a chatbot to answer your question regarding the Unique Fire 227 Engineered Manual.
\nNotes: This chatbot does not implement the memory module, ask specific questions and don't expect it remember the last conversation.
\nThe Manual can be found at https://drive.google.com/file/d/1zvkg2z1l7K6hdzsou654fD8Zv0xBbJvd/view?usp=sharing"""
)
iface.launch(share=False)