update app.py

app.py

@@ -17,6 +17,7 @@ class inferSSCL():
         self.base_models = {}
         self.batch_data = {}
         self.test_data = []
+        self.output = []
     
     def load_vocab_pretrain(self, file_pretrain_vocab, file_pretrain_vec, pad_tokens=True):
         vocab2id = {'<pad>': 0}
@@ -225,9 +226,9 @@ class inferSSCL():
     
     def get_aspect(self, pred, ignore='none'):
         if len(pred) > 1:
-            return(pred[1:])
+            self.output.append(pred[1:])
         else:
-            return(['None'])
+            self.output.append(['None'])
             
     def infer(self, text=''):
         self.args.task = 'sscl-infer'
@@ -265,7 +266,8 @@ args = parser.parse_args(args=[])
 model = inferSSCL(args)
 
 cmt = st.text_area('Enter some text: ')
-output = model.infer(cmt)
+model.infer(cmt)
 
+output = model.output[0]
 if output:
     st.title(output)