Update app.py

app.py

@@ -22,13 +22,12 @@ import nltk
 nltk.download('stopwords')
 texts=["nokia"]
 #print(make_prediction(texts))
-def make_prediction(texts):
-        st.write(texts[0])        
-        data,df=entity_extractor.input_text(texts)
-        probs = entity_extractor.extract_entity_probabilities( dataset=data)
-        for i in range(len(probs)):
-          prediction="Predicted Company Ticker: \n"+str(list(LABEL_MAP.keys())[list(LABEL_MAP.values()).index(np.argmax(probs[i]))])+'\n'
-        return prediction
+
 texts=["Nokia"]
-pred=make_prediction(texts)
-st.write(pred)
+st.write(texts[0])        
+data,df=entity_extractor.input_text(texts)
+probs = entity_extractor.extract_entity_probabilities( dataset=data)
+for i in range(len(probs)):
+  prediction="Predicted Company Ticker: \n"+str(list(LABEL_MAP.keys())[list(LABEL_MAP.values()).index(np.argmax(probs[i]))])+'\n'
+
+st.write(prediction)