upd code

app.py

@@ -9,8 +9,7 @@ st.markdown("### Privet, mir!")
 st.markdown("<img width=200px src='https://i.pinimg.com/736x/11/33/19/113319f0ffe91f4bb0f468914b9916da.jpg'>", unsafe_allow_html=True)
 
 text = st.text_area("ENTER TEXT HERE")
-t2 = text.upper()
-st.markdown(f"{t2}")
+st.markdown("INFERENCE STARTS ...")
 
 # dict for decoding / enclding labels
 labels = {'cs.NE': 0, 'cs.CL': 1, 'cs.AI': 2, 'stat.ML': 3, 'cs.CV': 4, 'cs.LG': 5}
@@ -37,8 +36,10 @@ class BertClassifier(nn.Module):
         return final_layer
         
 model = BertClassifier(n_classes=len(labels))
-model.load_state_dict(torch.load('model_weights_1.pt'))
+st.markdown("Model created")
+model.load_state_dict(torch.load('model_weights_1.pt', map_location=torch.device('cpu')))
 model.eval()
+st.markdown("Model weights loaded")
 
 def inference(txt, mode=None):
     # infers classes for text topic based on the trained model from above
@@ -70,5 +71,6 @@ def inference(txt, mode=None):
     else:
         return res   
         
-res = inference(text, mode=None)        
+res = inference(text, mode=None)    
+st.markdown("INFERENCE RESULT:")    
 st.markdown(f"{res}")