Update app.py

app.py

@@ -25,6 +25,7 @@ class_names = ['CRVO',
 resnet50, resnet50_transforms = create_resnet50_model(
     num_classes=len(class_names), # actual value would also work
 )
+model.fc = nn.Linear(2048, 10)
 
 # Load saved weights
 resnet50.load_state_dict(