dilbert img

app.py

@@ -6,6 +6,7 @@ from time import time
 import matplotlib.pyplot as plt
 # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 device = 'cpu'
+from PIL import Image
 
 # dict for decoding / enclding labels
 labels = {'cs.NE': 0, 'cs.CL': 1, 'cs.AI': 2, 'stat.ML': 3, 'cs.CV': 4, 'cs.LG': 5}
@@ -58,17 +59,19 @@ def inference(txt, mode=None):
     
 model = build_model()
     
-st.markdown("### Privet, mir!")
-st.markdown("<img width=200px src='https://i.pinimg.com/736x/11/33/19/113319f0ffe91f4bb0f468914b9916da.jpg'>", unsafe_allow_html=True)
+st.markdown("###Predict topic by abstract.")
+image = Image.open('dilbert_big_data.jpg')
+st.image(image)
+# 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")
+
 start_time = time()
-st.markdown("INFERENCE STARTS ...")
-    
 
 res = inference(text, mode=None)
 res.sort(key = lambda x : - x[1])
-st.markdown("<b>INFERENCE RESULT:</b>")    
+
+st.markdown("INFERENCE RESULT")    
 for lbl, score in res:
     if score >=1:
         st.markdown(f"[        {lbl:<7}] {labels_decoder[lbl]:<35}  {score:.1f}%")