app.py

'''
model_selector = '0'
queries_selector = '0'
prompt_selector = '0'
metric_selector = '0'
prediction_strategy_selector = ['1','2.0','2.1','2.2','2.3','3.0','3.1','3.2','3.3']
'''

import gradio as gr
import run
import options as op

#from evaluate.utils.logging import get_logger
#logger = get_logger(__name__)

iface = gr.Interface(
	run.tp_tf_test,
	[
		gr.Radio(list(op.models.keys())),
		gr.Radio(list(op.queries.keys())),
		gr.Radio(list(op.prompts.keys())),
		gr.Radio(list(op.metrics.keys())),
		gr.CheckboxGroup(list(op.prediction_strategy_options.keys())),
	],
	"text",
	title="Classification by SDGs",
	description="Zero Shot Classification by SDG over test dataset",
    	)

iface.launch()