Spaces:
Runtime error
Runtime error
File size: 935 Bytes
b3972fa dcf544a b3972fa dcf544a b2a42bc b3972fa b2a42bc 5c40ef7 b3972fa b2a42bc b3972fa b2a42bc b3972fa e21ecdd b3972fa b2a42bc b3972fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
from autollm import AutoQueryEngine
import os
import spaces
@spaces.GPU
def query_engine(llm_model, document, query):
api_key = os.getenv("HUGGINGFACE_API_KEY")
llm_api_base = "https://api-inference.huggingface.co/models/"
content = str(document)
query_engine = AutoQueryEngine.from_defaults(
documents=[content],
llm_model=llm_model,
llm_api_base=llm_api_base,
)
response = query_engine.query(query)
return response
interface = gr.Blocks()
with interface:
gr.Markdown("# AutoQueryEngine Interface")
llm_model = gr.Textbox(label="LLM Model", value="mistralai/Mixtral-8x7B-Instruct-v0.1")
document = gr.File(label="Documents")
query = gr.Textbox(label="Query")
output = gr.Textbox(label="Output")
query_btn = gr.Button("Query")
query_btn.click(fn=query_engine, inputs=[llm_model, document, query], outputs=output)
interface.launch() |