Spaces:

dwb2023
/

model_explorer2

Paused

dwb2023 commited on Jun 2, 2024

Commit

c26a148

verified ·

1 Parent(s): 51dd1fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModel
 def get_model_summary(model_name):
     model = AutoModel.from_pretrained(model_name)
     return str(model)

 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)