update app

app.py

@@ -246,12 +246,12 @@ def benchmark_interface(model_name):
     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]
@@ -275,8 +275,6 @@ iface = gr.Interface(
     outputs=[
         gr.Plot(label="Model Benchmark Results")
     ],
-    title="FasterAI Model Benchmark",
-    description="Select a pre-trained PyTorch model to visualize its performance metrics."
 )
 
 iface.launch()