kubrabuzlu commited on
Commit
93bc389
·
verified ·
1 Parent(s): 5187b63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +40 -40
app.py CHANGED
@@ -1,40 +1,40 @@
1
- import streamlit as st
2
-
3
- import os
4
-
5
- from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
6
- from langchain_community.vectorstores import FAISS
7
-
8
- st.set_page_config(page_title="Educate Kids", page_icon=":robot:")
9
- st.header("Hey, Ask me something & I will give out similar things")
10
-
11
- embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
12
-
13
- from langchain.document_loaders.csv_loader import CSVLoader
14
-
15
- loader = CSVLoader(file_path="myData.csv",
16
- csv_args={
17
- 'delimiter': ',',
18
- 'quotechar': '"',
19
- 'fieldnames': ['Words']
20
- })
21
-
22
- data = loader.load()
23
-
24
- print(data)
25
-
26
- db = FAISS.from_documents(data, embeddings)
27
-
28
- def get_text():
29
- input_text = st.text_input("You: ", key=input)
30
- return input_text
31
-
32
- user_input = get_text()
33
- submit = st.button("Find similar things")
34
-
35
- if submit:
36
- docs = db.similarity_search(user_input)
37
- print(docs)
38
- st.subheader("Top Matches: ")
39
- st.text(docs[0])
40
- st.text(docs[1].page_content)
 
1
+ import streamlit as st
2
+
3
+ import os
4
+
5
+ from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
6
+ from langchain_community.vectorstores import FAISS
7
+
8
+ st.set_page_config(page_title="Educate Kids", page_icon=":robot:")
9
+ st.header("Hey, Ask me something & I will give out similar things")
10
+
11
+ embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
12
+
13
+ from langchain.document_loaders.csv_loader import CSVLoader
14
+
15
+ loader = CSVLoader(file_path="myData.csv",
16
+ csv_args={
17
+ 'delimiter': ',',
18
+ 'quotechar': '"',
19
+ 'fieldnames': ['Words']
20
+ })
21
+
22
+ data = loader.load()
23
+
24
+ print(data)
25
+
26
+ db = FAISS.from_documents(data, embeddings)
27
+
28
+ def get_text():
29
+ input_text = st.text_input("You: ", key=input)
30
+ return input_text
31
+
32
+ user_input = get_text()
33
+ submit = st.button("Find similar things")
34
+
35
+ if submit:
36
+ docs = db.similarity_search(user_input)
37
+ print(docs)
38
+ st.subheader("Top Matches: ")
39
+ st.text(docs[0])
40
+ st.text(docs[1])