MostoHF commited on
Commit
d5d91e5
·
verified ·
1 Parent(s): e4ba362

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from transformers import AutoModel, AutoTokenizer
6
  from huggingface_hub import hf_hub_download
7
  from model import ClassificationModel
8
 
 
9
 
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
11
 
@@ -76,7 +77,6 @@ def inference(title, abstract, threshold=0.95):
76
 
77
  # ------------------- Streamlit UI -------------------
78
 
79
- st.set_page_config(page_title="Article Theme Classifier", layout="centered")
80
  st.title("📄 Article Theme Classifier")
81
 
82
  title = st.text_input("Title", value="Введите title...")
 
6
  from huggingface_hub import hf_hub_download
7
  from model import ClassificationModel
8
 
9
+ st.set_page_config(page_title="Article Theme Classifier", layout="centered")
10
 
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
 
 
77
 
78
  # ------------------- Streamlit UI -------------------
79
 
 
80
  st.title("📄 Article Theme Classifier")
81
 
82
  title = st.text_input("Title", value="Введите title...")