Spaces:
Sleeping
Sleeping
update model to setswana
Browse files
app.py
CHANGED
|
@@ -2,11 +2,11 @@ import gradio as gr
|
|
| 2 |
|
| 3 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
| 4 |
|
| 5 |
-
MODEL_URL = "https://huggingface.co/
|
| 6 |
WEBSITE_URL = "https://www.kodiks.com/ai_solutions.html"
|
| 7 |
|
| 8 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 9 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
| 10 |
|
| 11 |
def prediction(news):
|
| 12 |
# create pipeline
|
|
@@ -23,7 +23,7 @@ def prediction(news):
|
|
| 23 |
|
| 24 |
gradio_ui = gr.Interface(
|
| 25 |
fn=prediction,
|
| 26 |
-
title="
|
| 27 |
description=f"Enter Turkish news article to see the category of the news.\n For this classification, the {MODEL_URL} model was used.",
|
| 28 |
examples=[
|
| 29 |
['Son dakika haberi... Sağlık Bakanlığı corona virüsü (koronavirüs) için son gelişmeleri ve Türkiye Günlük Korona Tablosu\'nu açıkladı. Türkiye\'de son 24 saatte 355 bin 947 Kovid-19 testi yapıldı, 26 bin 896 kişinin testi pozitif çıktı, 210 kişi yaşamını yitirdi. Tabloda vaka ve vefat sayılarına dikkat çeken Bakan Koca, "Bu ölümlerin bir kısmı birden çok sebebin bir araya gelmesine bağlı ve önlenebilir. Örneğin, Covid-19’a kronik hastalıkların eşlik etmesi riski artıran bir faktör. Tam doz aşı bu riski azaltıyor." çağrısında bulundu. Öte yandan ikinci doz aşılama oranı %75\'e çıkan Tunceli, mavi kategoriye yükseldi.'],
|
|
|
|
| 2 |
|
| 3 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
| 4 |
|
| 5 |
+
MODEL_URL = "https://huggingface.co/dsfsi/PuoBERTa-News"
|
| 6 |
WEBSITE_URL = "https://www.kodiks.com/ai_solutions.html"
|
| 7 |
|
| 8 |
+
tokenizer = AutoTokenizer.from_pretrained("dsfsi/PuoBERTa-News")
|
| 9 |
+
model = AutoModelForSequenceClassification.from_pretrained("dsfsi/PuoBERTa-News")
|
| 10 |
|
| 11 |
def prediction(news):
|
| 12 |
# create pipeline
|
|
|
|
| 23 |
|
| 24 |
gradio_ui = gr.Interface(
|
| 25 |
fn=prediction,
|
| 26 |
+
title="Setswana News Classification",
|
| 27 |
description=f"Enter Turkish news article to see the category of the news.\n For this classification, the {MODEL_URL} model was used.",
|
| 28 |
examples=[
|
| 29 |
['Son dakika haberi... Sağlık Bakanlığı corona virüsü (koronavirüs) için son gelişmeleri ve Türkiye Günlük Korona Tablosu\'nu açıkladı. Türkiye\'de son 24 saatte 355 bin 947 Kovid-19 testi yapıldı, 26 bin 896 kişinin testi pozitif çıktı, 210 kişi yaşamını yitirdi. Tabloda vaka ve vefat sayılarına dikkat çeken Bakan Koca, "Bu ölümlerin bir kısmı birden çok sebebin bir araya gelmesine bağlı ve önlenebilir. Örneğin, Covid-19’a kronik hastalıkların eşlik etmesi riski artıran bir faktör. Tam doz aşı bu riski azaltıyor." çağrısında bulundu. Öte yandan ikinci doz aşılama oranı %75\'e çıkan Tunceli, mavi kategoriye yükseldi.'],
|