Update app.py
Browse files
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...")
|