Update app.py

app.py

@@ -1,18 +1,18 @@
 import streamlit as st
-# import torch
-# from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 
-# def combine_title_summary(title, summary):
-#     return "title: " + title + " summary: " + summary
+def combine_title_summary(title, summary):
+    return "title: " + title + " summary: " + summary
 
 
-# tag2ind = {
-#     "bio": 0,
-#     "physics": 1,
-#     "math": 2,
-#     "cs": 3,
-# }
+tag2ind = {
+    "bio": 0,
+    "physics": 1,
+    "math": 2,
+    "cs": 3,
+}
 
 
 # @st.cache_resource