Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
-
 @st.cache_resource
 def load_model():
     checkpoint = torch.load('TinyBERT_cls_model.pt', map_location='cpu')
@@ -43,7 +42,6 @@ def predict(title, abstract, model, tokenizer, idx_to_category, threshold=0.95):
     
     return results, cumulative_prob
 
-
 def main():
     model, tokenizer, idx_to_category = load_model()
     st.set_page_config(page_title="arXiv Classifier", layout="wide")
@@ -82,6 +80,5 @@ def main():
             with col2:
                 st.progress(res['probability'], text=res['category'])
 
-
 if __name__ == "__main__":
     main()