Commit
·
80989ca
1
Parent(s):
c1da9c3
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ tokenizer_name = AutoTokenizer.from_pretrained("ikoghoemmanuell/finetuned_fake_n
|
|
11 |
@st.cache_resource
|
12 |
def detect_fake_news(text):
|
13 |
# Load the pipeline.
|
14 |
-
|
15 |
|
16 |
# Predict the sentiment.
|
17 |
prediction = pipeline(text)
|
|
|
11 |
@st.cache_resource
|
12 |
def detect_fake_news(text):
|
13 |
# Load the pipeline.
|
14 |
+
pipeline = transformers.pipeline("text-classification", model=model_name, tokenizer=tokenizer_name)
|
15 |
|
16 |
# Predict the sentiment.
|
17 |
prediction = pipeline(text)
|