Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -18,7 +18,7 @@ using a fine-tuned `microsoft/deberta-v3-large` model.
|
|
| 18 |
# ====================== LOAD MODEL ======================
|
| 19 |
@st.cache_resource
|
| 20 |
def load_model():
|
| 21 |
-
model_name = "
|
| 22 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 23 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 24 |
return tokenizer, model
|
|
|
|
| 18 |
# ====================== LOAD MODEL ======================
|
| 19 |
@st.cache_resource
|
| 20 |
def load_model():
|
| 21 |
+
model_name = "kishankachhadiya/ai-text-detector-v-45.5" # Replace with your HF model repo
|
| 22 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 23 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 24 |
return tokenizer, model
|