Docfile commited on
Commit
e961a39
·
1 Parent(s): 217e409

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +54 -0
app.py ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ import sys
3
+ import gradio as gr
4
+ import asyncio
5
+ import nest_asyncio
6
+
7
+ logging.basicConfig(stream=sys.stdout, level=logging.INFO)
8
+ logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
9
+
10
+ from llama_index import VectorStoreIndex, SimpleDirectoryReader, ServiceContext
11
+ from llama_index.llms import HuggingFaceLLM
12
+ from langchain.embeddings import HuggingFaceEmbeddings, HuggingFaceInstructEmbeddings
13
+
14
+ from g4f import Provider, models
15
+ from langchain.llms.base import LLM
16
+ from llama_index.llms import LangChainLLM
17
+ from langchain_g4f import G4FLLM
18
+
19
+ nest_asyncio.apply()
20
+ """
21
+ documents = SimpleDirectoryReader('data').load_data()
22
+ model_kwargs = {'device': 'cpu'}
23
+ encode_kwargs = {'normalize_embeddings': True}
24
+ embed_model = HuggingFaceInstructEmbeddings(
25
+ model_name="hkunlp/instructor-large", model_kwargs=model_kwargs,
26
+ encode_kwargs=encode_kwargs
27
+ )
28
+ """
29
+ llm= LLM = G4FLLM(
30
+ model=models.gpt_35_turbo,
31
+ provider=Provider.FreeGpt,
32
+ )
33
+
34
+ llm = LangChainLLM(llm=llm)
35
+ #embed_model=embed_model)
36
+
37
+ service_context = ServiceContext.from_defaults(chunk_size=512, llm=llm)
38
+
39
+ from llama_index import StorageContext, load_index_from_storage
40
+
41
+ # rebuild storage context
42
+ storage_context = StorageContext.from_defaults(persist_dir="./storage")
43
+ # load index
44
+ index = load_index_from_storage(storage_context, service_context =service_context)
45
+
46
+ async def main(query):
47
+
48
+ query_engine = index.as_query_engine()
49
+ response = query_engine.query(query)
50
+ print(response)
51
+ return response
52
+
53
+ iface = gr.Interface(fn=main, inputs="text", outputs="text")
54
+ iface.launch()