update model file

model.py

@@ -39,5 +39,5 @@ def create_vit_model(num_classes:int=28,
     model.heads = torch.nn.Sequential(
     nn.Linear(in_features=768,
               out_features=28, # Number of Arabic letters = our classes
-              bias=True).to(device))
+              bias=True))
     return model, transform_pipeline