gpu

app.py

@@ -172,7 +172,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Row():
                 model_name = gr.Dropdown(
-                    choices=[str(model) for model in models],
+                    choices=[str(model_info) for model_info in models_info],
                     container=False
                 )
             load_button = gr.Button("Load Model")