Spaces:
Runtime error
Runtime error
| 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() | |