Update main.py

main.py

@@ -6,7 +6,7 @@ import io
 
 app = FastAPI()
 
-# Load the pre-trained VGG16 model for the prediction
+# Load the pre-trained VGG16 model for the prediction of model
 model = models.vgg16()
 num_features_in = model.classifier[6].in_features
 model.classifier[6] = torch.nn.Linear(num_features_in, 1)