Spaces:
Runtime error
Runtime error
File size: 1,744 Bytes
176fea5 1c26ac1 934a29a 797da9c 543da94 934a29a 1fc23de 934a29a 808f967 934a29a 543da94 2a450ee ed7b2a1 79e71b7 797da9c c618b89 ed7b2a1 86ebbba 23f332a 4d8977a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import gradio as gr
from gradio import FlaggingCallback
from gradio.components import IOComponent
from typing import List, Optional, Any
import os
import argilla as rg
class ArgillaLogger(FlaggingCallback):
def __init__(self, api_url, api_key):
rg.init(api_url=api_url, api_key=api_key)
def setup(self, components: List[IOComponent], flagging_dir: str):
pass
def flag(
self,
flag_data: List[Any],
flag_option: Optional[str] = None,
flag_index: Optional[int] = None,
username: Optional[str] = None,
) -> int:
text = flag_data[0]
inference = flag_data[1]
prediction = [(pred["label"], pred["confidence"]) for pred in inference["confidences"]]
rg.log(
name="cyber_bullying_sentiment_feedback",
records=rg.TextClassificationRecord(text=text, prediction=prediction)
)
description = "Get sentiment if a comment is 'no cyber bullying' or 'cyber bullying'.The predicted outputs refer to 'cyber bullying' for racism and sexism comments and 'no cyber bullying' for others comments.please do flag if you feel that the required output isn't desired output "
title = "Cyber Bullying Sentiment Classifier"
examples = [["lets have a coffee later on"],["go to hell Muslims "],["I respect your work ethics"]]
gr.Interface.load("models/Amitesh007/tw-sentiment-finetuned",
description = description,
title = title,
examples = examples,
allow_flagging="manual",
flagging_callback=ArgillaLogger(api_url="https://amitesh007-amitesh-argilla.hf.space", api_key="team.apikey")
).launch() |