update

app.py

@@ -143,7 +143,6 @@ def main_interface(model_name, sparsity, action):
         'MobileNetV2': models.mobilenet_v2(pretrained=False),
         'EfficientNet-B0': models.efficientnet_b0(pretrained=False),
         'VGG16': models.vgg16(pretrained=False),
-        'DenseNet121': models.densenet121(pretrained=False)
     }
     
     model = model_mapping[model_name]
@@ -166,7 +165,7 @@ def main_interface(model_name, sparsity, action):
             os.remove(temp_path)
 
 
-available_models = ['ResNet18', 'ResNet50', 'MobileNetV2', 'EfficientNet-B0', 'VGG16', 'DenseNet121']
+available_models = ['ResNet18', 'ResNet50', 'MobileNetV2', 'EfficientNet-B0', 'VGG16']
 
 iface = gr.Interface(
     fn=main_interface,