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...")
|