Spaces:
Sleeping
Sleeping
Commit
·
22956f8
1
Parent(s):
4dd0f5b
requirement.txt
Browse files- app.py +2 -4
- requirements.txt +4 -0
app.py
CHANGED
@@ -1,14 +1,12 @@
|
|
1 |
-
import numpy as np
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
import pinecone
|
5 |
import time
|
6 |
from torch import cuda
|
7 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
8 |
-
import PyPDF2
|
9 |
-
import re
|
10 |
from langchain.vectorstores import Pinecone
|
11 |
-
import os
|
12 |
|
13 |
embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
|
14 |
device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
|
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
import pinecone
|
4 |
import time
|
5 |
from torch import cuda
|
6 |
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
|
7 |
+
# import PyPDF2
|
8 |
+
# import re
|
9 |
from langchain.vectorstores import Pinecone
|
|
|
10 |
|
11 |
embed_model_id = 'sentence-transformers/all-MiniLM-L6-v2'
|
12 |
device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
|
requirements.txt
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
gradio
|
2 |
+
pinecone
|
3 |
+
torch
|
4 |
+
langchain
|