|
|
|
"""Application.ipynb |
|
|
|
Automatically generated by Colaboratory. |
|
|
|
Original file is located at |
|
https://colab.research.google.com/drive/148du8431_JkTaH-totdocC2aUXzOWimL |
|
""" |
|
|
|
import gradio as gr |
|
from transformers import BertTokenizer, TFBertForSequenceClassification |
|
import tensorflow as tf |
|
|
|
|
|
|
|
tokenizer = BertTokenizer.from_pretrained("nlpaueb/bert-base-greek-uncased-v1") |
|
|
|
|
|
model = TFBertForSequenceClassification.from_pretrained('Kleo/finetuned_bert') |
|
|
|
def check_sarcasm(sentence): |
|
tf_batch = tokenizer(sentence, max_length=128, padding=True, truncation=True, return_tensors='tf') |
|
tf_outputs = model(tf_batch.input_ids, tf_batch.token_type_ids) |
|
tf_predictions = tf.nn.softmax(tf_outputs.logits, axis=-1) |
|
pred_label = tf.argmax(tf_predictions, axis=1) |
|
|
|
if pred_label == 1: |
|
return "Sarcastic" |
|
else: |
|
return "Not sarcastic" |
|
|
|
|
|
|
|
iface = gr.Interface( |
|
fn=check_sarcasm, |
|
inputs="text", |
|
outputs="text", |
|
title="Sarcasm Detection", |
|
server_name="0.0.0.0", |
|
description="Enter a headline from the Greek news and check if it's sarcastic." |
|
) |
|
|
|
|
|
iface.launch() |
|
|
|
|
