Update app.py

app.py

@@ -18,9 +18,6 @@ class Net(nn.Module):
         return self.layer(x)
 
 def get_hidden_states(encoded, model):
-     """Push input IDs through model. Stack and sum `layers` (last four by default).
-        Select only those subword token outputs that belong to our word of interest
-        and average them."""
      with torch.no_grad():
          output = model(decoder_input_ids=encoded['input_ids'], output_hidden_states=True, **encoded)
 
@@ -63,12 +60,12 @@ sum = 0
 
 res = ''
 
+st.write('best tags:')
+
 for tag in best_tags:
   if sum > 0.95:
     break
   sum += logits[tag.item()]
   # print(tag.item())
   new_tag = labels_articles[tag.item() + 1]
-  res += new_tag + '\n'
-
-st.write('best tags = \n', res)
+  st.write(new_tag)