model_explorer2 / app.py
dwb2023's picture
AutoModel test
6bf2756 verified
raw
history blame
261 Bytes
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()