chatgptFinetune / app.py
mesutt's picture
Update app.py
ef58da7
raw
history blame contribute delete
785 Bytes
import llama_index
from llama_index import SimpleDirectoryReader, GPTVectorStoreIndex ,StorageContext,load_index_from_storage
import os
import gradio as gr
os.environ["OPENAI_API_KEY"] = "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"
storage_context = StorageContext.from_defaults(persist_dir="mustafaKemalAtatürk.txt")
index = load_index_from_storage(storage_context)
#prompt = "What is this text about"
#result = index.as_query_engine().query(prompt)
def query(prompt):
result = index.as_query_engine().query(prompt)
return result
with gr.Blocks() as demo:
prompt = gr.Textbox(label="Questions or query", placeholder="what is your query")
output = gr.Textbox()
query_btn = gr.Button("query")
query_btn.click(query, prompt, output)
demo.launch()