import transformers
import gradio as gr
import datasets
import torch
from diffusers import StableDiffusionPipeline
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import ImageGrid