Update app.py

app.py

@@ -26,7 +26,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)
+resnet50.fc = nn.Linear(2048, 10)
 
 # Load saved weights
 resnet50.load_state_dict(