karthickg12 commited on
Commit
bb48749
1 Parent(s): eed818f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,16 +16,16 @@
16
  # t1=st.text_input("Enter the Text for Summarization")
17
  # st.write(summarizer(t1))
18
 
19
- from transformers import AutoTokenizer, AutoModel
20
  import streamlit as st
21
 
22
- tokenizer = AutoTokenizer.from_pretrained("llmware/industry-bert-insurance-v0.1")
23
 
24
- model = AutoModel.from_pretrained("llmware/industry-bert-insurance-v0.1")
25
  # Use a pipeline as a high-level helper
26
  from transformers import pipeline
27
 
28
- pipe = pipeline("feature-extraction", model="llmware/industry-bert-insurance-v0.1")
29
 
30
  t=st.text_input("Enter the Text")
31
  st.write(pipe(t))
 
16
  # t1=st.text_input("Enter the Text for Summarization")
17
  # st.write(summarizer(t1))
18
 
19
+ #from transformers import AutoTokenizer, AutoModel
20
  import streamlit as st
21
 
22
+ #tokenizer = AutoTokenizer.from_pretrained("llmware/industry-bert-insurance-v0.1")
23
 
24
+ #model = AutoModel.from_pretrained("llmware/industry-bert-insurance-v0.1")
25
  # Use a pipeline as a high-level helper
26
  from transformers import pipeline
27
 
28
+ pipe = pipeline("feature-extraction")
29
 
30
  t=st.text_input("Enter the Text")
31
  st.write(pipe(t))