Update app.py

app.py

@@ -21,7 +21,7 @@ entity_extractor = BertEntityExtractor.load_trained_model()
 import nltk
 nltk.download('stopwords')
 texts=["nokia"]
-print(make_prediction(texts))
+#print(make_prediction(texts))
 def make_prediction(texts):        
         data,df=entity_extractor.input_text(texts)
         probs = entity_extractor.extract_entity_probabilities( dataset=data)
@@ -30,4 +30,4 @@ def make_prediction(texts):
         return prediction
 texts=["Nokia"]
 pred=make_prediction(texts)
-#st.write()
+st.write(pred)