Update app.py

app.py

@@ -3,27 +3,21 @@ import os
 import git
 if(os.path.isdir("./TickerExtraction")==False):
   git.Git("./").clone("https://huggingface.co/rajaatif786/TickerExtraction")
-  st.write("completed")
 print(os.path.exists("rajaatif786/TickerExtraction/entity_model2.pt"))
 st.write(os.listdir("./TickerExtraction/"))
-import streamlit as st
-
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
 
-import os
 import pandas as pd
 import numpy as np
-#device="cpu"
 os.chdir("./TickerExtraction")
 from EntityExtractor import EntityDataset, EntityBertNet,BertEntityExtractor, LABEL_MAP
 entity_extractor = BertEntityExtractor.load_trained_model()
 import nltk
 nltk.download('stopwords')
-texts=["nokia"]
-#print(make_prediction(texts))
 
-texts=["Nokia"]
+
+texts=[st.text_input("Enter Text")]
 st.write(texts[0])        
 data,df=entity_extractor.input_text(texts)
 probs = entity_extractor.extract_entity_probabilities( dataset=data)