Spaces:
Runtime error
Runtime error
import ktrain | |
from gradio import Interface, Parallel, TabbedInterface | |
vs_examples = [ | |
["I only get my kids the ones I got....I've turned down many so called 'vaccines'"], | |
["In child protective services, further providing for definitions, for immunity from liability"], | |
["Lol what? Measles is a real thing. Get vaccinated"]] | |
vs_title = "Vaccine Sentiment Task - VS2" | |
vs_desc = "Enter vaccine-related tweets to generate sentiment from 3 models (BERT, MentalBERT, PHS-BERT). Label 0='vaccine critical', 1='neutral', 2='vaccine supportive'. The three provided examples have true labels 0,1,2 respectively. For details about VS2, please refer to our paper (linked provided in the corresponding Hugging Face repository)." | |
vs_predictor_bert = ktrain.load_predictor('vs/bert') | |
vs_predictor_mental = ktrain.load_predictor('vs/mentalbert') | |
vs_predictor_phs = ktrain.load_predictor('vs/phsbert') | |
def vs_BERT(text): | |
results = vs_predictor_bert.predict(str(text)) | |
return str(results) | |
def vs_MentalBERT(text): | |
results = vs_predictor_mental.predict(str(text)) | |
return str(results) | |
def vs_PHSBERT(text): | |
results = vs_predictor_phs.predict(str(text)) | |
return str(results) | |
vs_bert_io = Interface(fn=vs_BERT, inputs="text", outputs="text") | |
vs_mental_io = Interface(fn=vs_MentalBERT, inputs="text", outputs="text") | |
vs_phs_io = Interface(fn=vs_PHSBERT, inputs="text", outputs="text") | |
vs = Parallel(vs_bert_io, vs_mental_io, vs_phs_io, | |
examples=vs_examples, | |
title=vs_title, | |
description=vs_desc, | |
theme="peach") | |
def model(text): | |
return "Predictions unavailable - to be completed." | |
hm = Interface(fn=model, inputs="text", outputs="text") | |
dep = Interface(fn=model, inputs="text", outputs="text") | |
covid = Interface(fn=model, inputs="text", outputs="text") | |
suicide = Interface(fn=model, inputs="text", outputs="text") | |
stress = Interface(fn=model, inputs="text", outputs="text") | |
other = Interface(fn=model, inputs="text", outputs="text") | |
interfaces = [vs, hm, dep, covid, suicide, stress, other] | |
interface_names = ["Vaccine Sentiment Task", | |
"Health Mention Task", | |
"Depression Task", | |
"COVID Related Task", | |
"Suicide Task", | |
"Stress Task", | |
"Other Health Related Task"] | |
TabbedInterface(interfaces, interface_names).launch() | |