update app

app.py

@@ -138,12 +138,12 @@ def main_interface(model_name, sparsity, action):
     import torchvision.models as models
     
     model_mapping = {
-        'ResNet18': models.resnet18(pretrained=True),
-        'ResNet50': models.resnet50(pretrained=True),
-        'MobileNetV2': models.mobilenet_v2(pretrained=True),
-        'EfficientNet-B0': models.efficientnet_b0(pretrained=True),
-        'VGG16': models.vgg16(pretrained=True),
-        'DenseNet121': models.densenet121(pretrained=True)
+        'ResNet18': models.resnet18(pretrained=False),
+        'ResNet50': models.resnet50(pretrained=False),
+        '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]
@@ -187,8 +187,6 @@ iface = gr.Interface(
         gr.File(label="Download Compressed Model"),
         gr.Plot(label="Size Comparison")  # Changed from gr.Image to gr.Plot
     ],
-    title="FasterAI Compressor",
-    description="Select a pre-trained PyTorch model to compress using our optimization techniques.",
 )
 
 iface.launch()