import gradio as gr
import torch
import torch.nn as nn
from transformers import XLMRobertaTokenizer, XLMRobertaModel
from huggingface_hub import hf_hub_download
import json