Commit
·
1094480
1
Parent(s):
757c949
Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,12 @@
|
|
1 |
import os
|
2 |
import streamlit as st
|
3 |
-
|
4 |
from datasets import load_dataset
|
5 |
import re
|
6 |
-
import boto3
|
7 |
import time
|
8 |
import requests
|
9 |
from bs4 import BeautifulSoup
|
10 |
import pandas as pd
|
11 |
-
import pinecone
|
12 |
import torch.nn.functional as F
|
13 |
from torch import Tensor
|
14 |
from transformers import AutoTokenizer, AutoModel
|
@@ -17,8 +15,6 @@ from keyphrase_vectorizers import KeyphraseCountVectorizer
|
|
17 |
kw_model=KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
18 |
|
19 |
|
20 |
-
pinecone.init(api_key=INDEX_API_KEY, environment=INDEX_ENV)
|
21 |
-
index = pinecone.Index(index_name=INDEX_NAME )
|
22 |
|
23 |
tokenizer = AutoTokenizer.from_pretrained('intfloat/e5-base')
|
24 |
model = AutoModel.from_pretrained('intfloat/e5-base')
|
|
|
1 |
import os
|
2 |
import streamlit as st
|
3 |
+
|
4 |
from datasets import load_dataset
|
5 |
import re
|
|
|
6 |
import time
|
7 |
import requests
|
8 |
from bs4 import BeautifulSoup
|
9 |
import pandas as pd
|
|
|
10 |
import torch.nn.functional as F
|
11 |
from torch import Tensor
|
12 |
from transformers import AutoTokenizer, AutoModel
|
|
|
15 |
kw_model=KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
16 |
|
17 |
|
|
|
|
|
18 |
|
19 |
tokenizer = AutoTokenizer.from_pretrained('intfloat/e5-base')
|
20 |
model = AutoModel.from_pretrained('intfloat/e5-base')
|