Spaces:
Sleeping
Sleeping
Update models/test2.py
Browse files- models/test2.py +2 -1
models/test2.py
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-strength")
|
|
|
5 |
target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
|
6 |
|
7 |
TR_TWEETS_EXAMPLES = [
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
#strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-strength")
|
5 |
+
strength_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/BERTurk-hate-speech-strength-prediction")
|
6 |
target_pipeline_tr = pipeline(task = "text-classification", model = "SoDehghan/test-target")
|
7 |
|
8 |
TR_TWEETS_EXAMPLES = [
|