Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
from transformers import pipeline, BertTokenizer
|
4 |
|
5 |
tokenizer = BertTokenizer.from_pretrained("indobenchmark/indobert-base-p1")
|
6 |
-
classifier = pipeline(model="
|
7 |
|
8 |
def predict(prompt):
|
9 |
completion = classifier(prompt)[0]
|
|
|
3 |
from transformers import pipeline, BertTokenizer
|
4 |
|
5 |
tokenizer = BertTokenizer.from_pretrained("indobenchmark/indobert-base-p1")
|
6 |
+
classifier = pipeline(model="umn-cyber/indobert-hoax", tokenizer=tokenizer)
|
7 |
|
8 |
def predict(prompt):
|
9 |
completion = classifier(prompt)[0]
|