Spaces:
Runtime error
Runtime error
File size: 853 Bytes
8d9bc51 60cb125 a2a3c2b 60cb125 8d9bc51 60cb125 8d9bc51 60cb125 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
import transformers
tokenizer = transformers.AutoTokenizer.from_pretrained("xlm-roberta-large")
model = transformers.AutoModelForSequenceClassification.from_pretrained("xlm-roberta-large", num_labels=2)
def predict(first_option, second_option):
input_ids = tokenizer.encode(first_option, second_option, return_tensors="pt", truncation=True, padding=True)
output = model(input_ids)[0]
result = torch.argmax(output)
return first_option if result == 0 else second_option
inputs = [gr.inputs.Textbox(label="Option 1"), gr.inputs.Textbox(label="Option 2")]
output = gr.outputs.Textbox(label="Chosen Option")
interface = gr.Interface(fn=predict, inputs=inputs, outputs=output, title="Decision Making with XLM-Roberta-Large", description="Input your two options and let XLM-Roberta-Large choose one.")
interface.launch()
|