import gradio as gr
from transformers import AutoModel

@spaces.GPU
def get_model_summary(model_name):
    model = AutoModel.from_pretrained(model_name)
    return str(model)

interface = gr.Interface(fn=get_model_summary, inputs="text", outputs="text")
interface.launch()