DeepSoft-Tech commited on
Commit
1094480
·
1 Parent(s): 757c949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -1,14 +1,12 @@
1
  import os
2
  import streamlit as st
3
- from get_pat_data import Patent_DataCreator
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')