File size: 377 Bytes
5c5b818
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from autollm import AutoQueryEngine
from llama_index import SimpleDirectoryReader
import gradio as gr

documents = SimpleDirectoryReader('data').load_data()

query_engine = AutoQueryEngine.from_parameters(
    documents = documents
)

def main(query):
    return query_engine.query(query).response

demo = gr.Interface(fn=main, inputs="text", outputs="text")
    
demo.launch()