File size: 261 Bytes
1efd233
6bf2756
1efd233
6bf2756
 
 
1efd233
6bf2756
 
1
2
3
4
5
6
7
8
9
10
import gradio as gr
from transformers import AutoModel

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()